BAAI
/

shunxing1234 commited on
Commit
3c2c00b
·
1 Parent(s): 77028aa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -41,7 +41,7 @@ The additional details of the Aquila model will be presented in the official tec
41
  from transformers import AutoTokenizer, AutoModelForCausalLM
42
  import torch
43
  device = torch.device("cuda")
44
- model_info = "BAAI/AquilaChat-7B"
45
  tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
46
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
47
  model.eval()
@@ -59,4 +59,4 @@ with torch.no_grad():
59
 
60
  ## License
61
 
62
- AquilaChat-7B and AquilaChat-33B open-source model is licensed under [ BAAI Aquila Model Licence Agreement](https://huggingface.co/BAAI/AquilaChat-7B/resolve/main/BAAI%20Aquila%20Model%20License%20Agreement.pdf)
 
41
  from transformers import AutoTokenizer, AutoModelForCausalLM
42
  import torch
43
  device = torch.device("cuda")
44
+ model_info = "BAAI/AquilaChat2-7B"
45
  tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
46
  model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
47
  model.eval()
 
59
 
60
  ## License
61
 
62
+ AquilaChat2-7B open-source model is licensed under [ BAAI Aquila Model Licence Agreement](https://huggingface.co/BAAI/AquilaChat-7B/resolve/main/BAAI%20Aquila%20Model%20License%20Agreement.pdf)