Update README
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ We can now run inference on this model:
|
|
| 53 |
|
| 54 |
```python
|
| 55 |
import torch
|
| 56 |
-
from transformers import AutoTokenizer,
|
| 57 |
|
| 58 |
# Load the tokenizer and model
|
| 59 |
model_path = "nvidia/Mistral-NeMo-Minitron-8B-Base"
|
|
|
|
| 53 |
|
| 54 |
```python
|
| 55 |
import torch
|
| 56 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 57 |
|
| 58 |
# Load the tokenizer and model
|
| 59 |
model_path = "nvidia/Mistral-NeMo-Minitron-8B-Base"
|