Salesforce
/

codegen2-1B_P

Text Generation

Inference Endpoints

Model card Files Files and versions Community

rooa commited on May 4, 2023

Commit

f37be90

•

1 Parent(s): 45fb58f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ The final snippet looks as follows:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen2-1B")
-model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen2-1B")
 def format(prefix, suffix):

 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen2-1B")
+model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen2-1B", trust_remote_code=True, revision="main")
 def format(prefix, suffix):