Update README.md
Browse files
README.md
CHANGED
@@ -107,7 +107,7 @@ Mainly playing around with tiny chat models - while the output is generally inta
|
|
107 |
```python
|
108 |
import torch
|
109 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
110 |
-
checkpoint = "LemiSt/SmolLM-135M-instruct-de"
|
111 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
112 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
113 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map=device, torch_dtype=torch.bfloat16)
|
|
|
107 |
```python
|
108 |
import torch
|
109 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
110 |
+
checkpoint = "LemiSt/SmolLM-135M-instruct-de-merged"
|
111 |
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
|
112 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
113 |
model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map=device, torch_dtype=torch.bfloat16)
|