Spaces:
Runtime error
Runtime error
Commit
·
d49742e
1
Parent(s):
42903b7
Update llama_chat_bot.py
Browse files- llama_chat_bot.py +1 -2
llama_chat_bot.py
CHANGED
@@ -14,14 +14,13 @@ import transformers
|
|
14 |
import os
|
15 |
|
16 |
hf_token=os.environ['HF_TOKEN']
|
17 |
-
|
18 |
print(hf_token)
|
19 |
|
20 |
|
21 |
from transformers import AutoTokenizer
|
22 |
|
23 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
24 |
-
tokenizer = AutoTokenizer.from_pretrained(model, token=hf_token)
|
25 |
|
26 |
# pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
27 |
|
|
|
14 |
import os
|
15 |
|
16 |
hf_token=os.environ['HF_TOKEN']
|
|
|
17 |
print(hf_token)
|
18 |
|
19 |
|
20 |
from transformers import AutoTokenizer
|
21 |
|
22 |
model = "meta-llama/Llama-2-7b-chat-hf"
|
23 |
+
tokenizer = await 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 |
|