Update README.md
Browse files
README.md
CHANGED
|
@@ -134,7 +134,7 @@ The SFT (Code) scores are excellent. The generalist scores (Math, Reasoning) are
|
|
| 134 |
import torch
|
| 135 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 136 |
|
| 137 |
-
model_id = "
|
| 138 |
device = "cuda" # or "cpu"
|
| 139 |
|
| 140 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
|
|
| 134 |
import torch
|
| 135 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 136 |
|
| 137 |
+
model_id = "Beebey/smallcoder-303m"
|
| 138 |
device = "cuda" # or "cpu"
|
| 139 |
|
| 140 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|