Update README_zh.md
Browse files- README_zh.md +1 -1
README_zh.md
CHANGED
|
@@ -50,7 +50,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 50 |
device = "cuda" # 加载模型的设备
|
| 51 |
|
| 52 |
model = AutoModelForCausalLM.from_pretrained(
|
| 53 |
-
"sii-research/InnoSpark-72B-
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto"
|
| 56 |
)
|
|
|
|
| 50 |
device = "cuda" # 加载模型的设备
|
| 51 |
|
| 52 |
model = AutoModelForCausalLM.from_pretrained(
|
| 53 |
+
"sii-research/InnoSpark-R-72B-0701",
|
| 54 |
torch_dtype="auto",
|
| 55 |
device_map="auto"
|
| 56 |
)
|