shobhitjethani commited on
Commit
d49742e
·
1 Parent(s): 42903b7

Update llama_chat_bot.py

Browse files
Files changed (1) hide show
  1. 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