Hugging Face's logo

BAAI
/
AquilaChat2-70B-Expr

Text Generation
Transformers
PyTorch
aquila
Model card Files
xet
 Community
1
ldwang commited on
Commit
0aecd41
·
1 Parent(s): 589784e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ import torch
26
  from transformers import AutoTokenizer, AutoModelForCausalLM
27
  from transformers import BitsAndBytesConfig
28
 
29
- model_info = "BAAI/AquilaChat2-70B"
30
  tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
31
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True, torch_dtype=torch.bfloat16)
32
  model.eval()
 
26
  from transformers import AutoTokenizer, AutoModelForCausalLM
27
  from transformers import BitsAndBytesConfig
28
 
29
+ model_info = "BAAI/AquilaChat2-70B-Expr"
30
  tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
31
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True, torch_dtype=torch.bfloat16)
32
  model.eval()