Update README.md
Browse files
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(
|
39 |
-
model = AutoModelForCausalLM.from_pretrained(
|
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-mono")
|
39 |
+
model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-16B-mono")
|
40 |
|
41 |
text = "def hello_world():"
|
42 |
input_ids = tokenizer(text, return_tensors="pt").input_ids
|