Update README.md
Browse files
README.md
CHANGED
|
@@ -55,7 +55,7 @@ We recommend using the same prompt in inference to obtain the best results!
|
|
| 55 |
Below you can find an example of model usage:
|
| 56 |
|
| 57 |
```python
|
| 58 |
-
from transformers import AutoTokenizer
|
| 59 |
import transformers
|
| 60 |
import torch
|
| 61 |
import os
|
|
|
|
| 55 |
Below you can find an example of model usage:
|
| 56 |
|
| 57 |
```python
|
| 58 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 59 |
import transformers
|
| 60 |
import torch
|
| 61 |
import os
|