shobhitjethani commited on
Commit
3237be6
1 Parent(s): d49742e

Update llama_chat_bot.py

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