Update README.md
Browse files
README.md
CHANGED
@@ -18,7 +18,7 @@ from transformers.generation import GenerationConfig
|
|
18 |
import os
|
19 |
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
20 |
|
21 |
-
model_path="yuyijiong/LongAlpaca-7b-chinese"
|
22 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
23 |
|
24 |
# use auto mode, automatically select precision based on the device.
|
|
|
18 |
import os
|
19 |
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
20 |
|
21 |
+
model_path="yuyijiong/LongAlpaca-7b-32k-chinese"
|
22 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
23 |
|
24 |
# use auto mode, automatically select precision based on the device.
|