Update README.md
Browse files
README.md
CHANGED
|
@@ -50,7 +50,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 50 |
device = "cuda" # the device to load the model onto
|
| 51 |
|
| 52 |
model = AutoModelForCausalLM.from_pretrained(
|
| 53 |
-
"sii-research/InnoSpark-72B-
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto"
|
| 56 |
)
|
|
|
|
| 50 |
device = "cuda" # the device to load the model onto
|
| 51 |
|
| 52 |
model = AutoModelForCausalLM.from_pretrained(
|
| 53 |
+
"sii-research/InnoSpark-R-72B-0701",
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto"
|
| 56 |
)
|