BAAI
/

ldwang commited on
Commit
b9d2f60
1 Parent(s): f891e7c

Update README_zh.md

Browse files
Files changed (1) hide show
  1. 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()