Add missing import to example in readme.
Browse files
README.md
CHANGED
|
@@ -54,7 +54,7 @@ TBA
|
|
| 54 |
### Example of usage 👩💻
|
| 55 |
```py
|
| 56 |
import torch
|
| 57 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer
|
| 58 |
|
| 59 |
model_id = "mrm8488/falcoder-7b"
|
| 60 |
|
|
|
|
| 54 |
### Example of usage 👩💻
|
| 55 |
```py
|
| 56 |
import torch
|
| 57 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, GenerationConfig
|
| 58 |
|
| 59 |
model_id = "mrm8488/falcoder-7b"
|
| 60 |
|