Spaces:
Runtime error
Runtime error
Commit
·
3df5cc4
1
Parent(s):
f152dfa
Update llama_chat_bot.py
Browse files- llama_chat_bot.py +3 -3
llama_chat_bot.py
CHANGED
@@ -7,8 +7,8 @@ Original file is located at
|
|
7 |
https://colab.research.google.com/drive/1BBWPVOOR0790ZALdTJy_pYV5bFf8KaGp
|
8 |
"""
|
9 |
# curl --proto "=https" --tlsv1.2 -sSf https://sh.rustup.rs | sh
|
10 |
-
pip install -e transformers torch accelerate
|
11 |
-
pip install -e --upgrade gradio
|
12 |
|
13 |
|
14 |
from transformers import AutoTokenizer
|
@@ -16,7 +16,7 @@ from transformers import AutoTokenizer
|
|
16 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
17 |
tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
|
18 |
|
19 |
-
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
20 |
|
21 |
from transformers import pipeline
|
22 |
|
|
|
7 |
https://colab.research.google.com/drive/1BBWPVOOR0790ZALdTJy_pYV5bFf8KaGp
|
8 |
"""
|
9 |
# curl --proto "=https" --tlsv1.2 -sSf https://sh.rustup.rs | sh
|
10 |
+
# pip install -e transformers torch accelerate
|
11 |
+
# pip install -e --upgrade gradio
|
12 |
|
13 |
|
14 |
from transformers import AutoTokenizer
|
|
|
16 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
17 |
tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
|
18 |
|
19 |
+
# pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
20 |
|
21 |
from transformers import pipeline
|
22 |
|