shunxing1234
commited on
Commit
•
7810f72
1
Parent(s):
984405c
Update README.md
Browse files
README.md
CHANGED
@@ -43,8 +43,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
43 |
import torch
|
44 |
|
45 |
model_info = "BAAI/Aquila-7B"
|
46 |
-
tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True
|
47 |
-
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True
|
48 |
model.eval()
|
49 |
model.to("cuda:0")
|
50 |
|
|
|
43 |
import torch
|
44 |
|
45 |
model_info = "BAAI/Aquila-7B"
|
46 |
+
tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
|
47 |
+
model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
|
48 |
model.eval()
|
49 |
model.to("cuda:0")
|
50 |
|