Update README.md

rooa · web-flow · commit 21b94e42faaf · 2023-07-06T12:42:12.000-07:00
diff --git a/codegen25/README.md b/codegen25/README.md
@@ -30,8 +30,8 @@ Program synthesis in the form of auto-regressive sampling can be performed as fo
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen25-7B-multi", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen25-7B-multi")
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen25-7b-mono", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen25-7b-mono")
 inputs = tokenizer("def hello_world():", return_tensors="pt")
 sample = model.generate(**inputs, max_length=128)
 print(tokenizer.decode(sample[0]))
