Commit
·
0e4a922
1
Parent(s):
02dc13e
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,9 @@ from peft import PeftModel
|
|
3 |
import transformers
|
4 |
import gradio as gr
|
5 |
|
|
|
|
|
|
|
6 |
assert (
|
7 |
"LlamaTokenizer" in transformers._import_structure["models.llama"]
|
8 |
), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
|
|
|
3 |
import transformers
|
4 |
import gradio as gr
|
5 |
|
6 |
+
from huggingface_hub import notebook_login
|
7 |
+
notebook_login()
|
8 |
+
|
9 |
assert (
|
10 |
"LlamaTokenizer" in transformers._import_structure["models.llama"]
|
11 |
), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
|