fix typo (#2)
Browse files- fix typo (723f6b57e6edf0462ae95f7657e0b36fe0a41113)
Co-authored-by: Julien Abadji <[email protected]>
README.md
CHANGED
|
@@ -30,7 +30,7 @@ pip install transformers
|
|
| 30 |
```python
|
| 31 |
# pip install transformers
|
| 32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 33 |
-
checkpoint = "HuggingFaceTB/SmolLM-360M-
|
| 34 |
|
| 35 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
|
|
|
| 30 |
```python
|
| 31 |
# pip install transformers
|
| 32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 33 |
+
checkpoint = "HuggingFaceTB/SmolLM-360M-Instruct"
|
| 34 |
|
| 35 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|