Spaces:
Runtime error
Runtime error
Commit
•
3237be6
1
Parent(s):
d49742e
Update llama_chat_bot.py
Browse files- llama_chat_bot.py +1 -1
llama_chat_bot.py
CHANGED
@@ -20,7 +20,7 @@ print(hf_token)
|
|
20 |
from transformers import AutoTokenizer
|
21 |
|
22 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
23 |
-
tokenizer =
|
24 |
|
25 |
# pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
26 |
|
|
|
20 |
from transformers import AutoTokenizer
|
21 |
|
22 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
23 |
+
tokenizer = AutoTokenizer.from_pretrained(model, token=os.environ['HF_TOKEN'])
|
24 |
|
25 |
# pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
26 |
|