Spaces:
Sleeping
Sleeping
Commit
·
a0b0102
1
Parent(s):
0b8b900
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import json
|
|
11 |
|
12 |
assert (
|
13 |
"LlamaTokenizer" in transformers._import_structure["models.llama"]
|
14 |
-
), "
|
15 |
from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
|
16 |
|
17 |
if torch.cuda.is_available():
|
@@ -26,7 +26,7 @@ except:
|
|
26 |
pass
|
27 |
|
28 |
|
29 |
-
base_model = "/
|
30 |
|
31 |
tokenizer = LlamaTokenizer.from_pretrained(base_model)
|
32 |
load_8bit = False
|
|
|
11 |
|
12 |
assert (
|
13 |
"LlamaTokenizer" in transformers._import_structure["models.llama"]
|
14 |
+
), "Please reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
|
15 |
from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
|
16 |
|
17 |
if torch.cuda.is_available():
|
|
|
26 |
pass
|
27 |
|
28 |
|
29 |
+
base_model = "https://huggingface.co/Shangding-Gu/Lunyu-LLM/"
|
30 |
|
31 |
tokenizer = LlamaTokenizer.from_pretrained(base_model)
|
32 |
load_8bit = False
|