Update README.md

README.md

@@ -84,7 +84,7 @@ pipeline_tag: text-generation
 
 
 <div style="display: flex; justify-content: center; align-items: center; flex-direction: column">
-    <h1 style="font-size: 5em; margin-bottom: 0; padding-bottom: 0;">MARS</h1>
+    <h1 style="font-size: 5em; margin-bottom: 0; padding-bottom: 0;">MARS-v0.2</h1>
     <aside>by <a href="https://curiosity.tech">Curiosity Technology</a></aside>
 </div>
 
@@ -111,7 +111,7 @@ You can run conversational inference using the Transformers pipeline abstraction
 import transformers
 import torch
 
-model_id = "curiositytech/MARS"
+model_id = "curiositytech/MARS-v0.2"
 
 pipeline = transformers.pipeline(
     "text-generation",
@@ -147,7 +147,7 @@ print(outputs[0]["generated_text"][-1])
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-model_id = "curiositytech/MARS"
+model_id = "curiositytech/MARS-v0.2"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(