Salesforce
/

codegen-16B-multi

Text Generation

Inference Endpoints

Model card Files Files and versions Community

rooa commited on Jun 28, 2022

Commit

f509e15

•

1 Parent(s): 37166ae

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-multi')
-model = AutoModelForCausalLM.from_pretrained('Salesforce/codegen-16B-multi')
 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-multi")
+model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-multi")
 text = "def hello_world():"
 input_ids = tokenizer(text, return_tensors="pt").input_ids