Spaces:
Runtime error
Runtime error
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
|