Update README_zh.md
Browse files- README_zh.md +1 -1
README_zh.md
CHANGED
@@ -30,7 +30,7 @@ import torch
|
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
31 |
from transformers import BitsAndBytesConfig
|
32 |
|
33 |
-
model_info = "BAAI/Aquila2-70B"
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
|
35 |
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
|
36 |
model.eval()
|
|
|
30 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
31 |
from transformers import BitsAndBytesConfig
|
32 |
|
33 |
+
model_info = "BAAI/Aquila2-70B-Expr"
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
|
35 |
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
|
36 |
model.eval()
|