Salesforce
/

codegen-16B-nl

Text Generation

Model card Files Files and versions

rooa commited on Jun 28, 2022

Commit

3b002d5

·

1 Parent(s): 5b043d8

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