Fixed model name in example (#1)
Browse files- Fixed model name in example (bc26f7e23b56d3f536cf8088f29c27422997e1fc)
Co-authored-by: Jaskirat Singh <[email protected]>
README.md
CHANGED
|
@@ -75,7 +75,7 @@ The final snippet looks as follows:
|
|
| 75 |
```python
|
| 76 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 77 |
tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen25-7b-multi", trust_remote_code=True)
|
| 78 |
-
model = AutoModelForCausalLM.from_pretrained("Salesforce/
|
| 79 |
|
| 80 |
|
| 81 |
def format(prefix, suffix):
|
|
|
|
| 75 |
```python
|
| 76 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 77 |
tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen25-7b-multi", trust_remote_code=True)
|
| 78 |
+
model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen25-7b-multi")
|
| 79 |
|
| 80 |
|
| 81 |
def format(prefix, suffix):
|