Salesforce
/

codegen-16B-mono

Text Generation

Inference Endpoints

Model card Files Files and versions Community

rooa commited on Jun 28, 2022

Commit

a214204

•

1 Parent(s): 19cf198

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -35,8 +35,8 @@ This model can be easily loaded using the `AutoModelForCausalLM` functionality:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained('Salesforce/codegen-16B-mono')
-model = AutoModelForCausalLM.from_pretrained('Salesforce/codegen-16B-mono')
 text = "def hello_world():"
 input_ids = tokenizer(text, return_tensors="pt").input_ids

 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-16B-mono")
+model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-mono")
 text = "def hello_world():"
 input_ids = tokenizer(text, return_tensors="pt").input_ids