shunxing1234
commited on
Commit
·
3c2c00b
1
Parent(s):
77028aa
Update README.md
Browse files
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/
|
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 |
-
|
|
|
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)
|