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