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

Update llama_chat_bot.py

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