Hugging Face's logo

Salesforce
/
codegen-16B-multi

Text Generation Transformers PyTorch codegen Inference Endpoints
Model card Files Community
3
rooa commited on
Commit
f509e15
•
1 Parent(s): 37166ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,8 +35,8 @@ This model can be easily loaded using the `AutoModelForCausalLM` functionality:
35
 
36
  ```python
37
  from transformers import AutoTokenizer, AutoModelForCausalLM
38
- tokenizer = AutoTokenizer.from_pretrained('Salesforce/codegen-16B-multi')
39
- model = AutoModelForCausalLM.from_pretrained('Salesforce/codegen-16B-multi')
40
 
41
  text = "def hello_world():"
42
  input_ids = tokenizer(text, return_tensors="pt").input_ids
35
 
36
  ```python
37
  from transformers import AutoTokenizer, AutoModelForCausalLM
38
+ tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen-16B-multi")
39
+ model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-multi")
40
 
41
  text = "def hello_world():"
42
  input_ids = tokenizer(text, return_tensors="pt").input_ids