Commit
•
f7310b3
1
Parent(s):
b6bdcd4
Fix add missing import
Browse files
README.md
CHANGED
|
@@ -169,7 +169,7 @@ Use the code below to get started with LINCE-ZERO!
|
|
| 169 |
|
| 170 |
```py
|
| 171 |
import torch
|
| 172 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
|
| 173 |
|
| 174 |
model_id = "clibrain/lince-zero"
|
| 175 |
|
|
|
|
| 169 |
|
| 170 |
```py
|
| 171 |
import torch
|
| 172 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, GenerationConfig
|
| 173 |
|
| 174 |
model_id = "clibrain/lince-zero"
|
| 175 |
|