Update README.md
Browse files
README.md
CHANGED
|
@@ -57,7 +57,7 @@ The final snippet looks as follows:
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen2-7B")
|
| 60 |
-
model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen2-7B")
|
| 61 |
|
| 62 |
|
| 63 |
def format(prefix, suffix):
|
|
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained("Salesforce/codegen2-7B")
|
| 60 |
+
model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen2-7B", trust_remote_code=True, revision="main")
|
| 61 |
|
| 62 |
|
| 63 |
def format(prefix, suffix):
|