shobhitjethani commited on
Commit
7f08f65
1 Parent(s): 40d0401

Update llama_chat_bot.py

Browse files
Files changed (1) hide show
  1. llama_chat_bot.py +5 -0
llama_chat_bot.py CHANGED
@@ -12,16 +12,21 @@ Original file is located at
12
 
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
 
27
  from transformers import pipeline
 
12
 
13
  import transformers
14
  import os
15
+ import huggingface_hub
16
 
17
  hf_token=os.environ['HF_READ_TOKEN']
18
  print(hf_token)
19
 
20
+ huggingface-cli login --token $HF_READ_TOKEN
21
+
22
 
23
  from transformers import AutoTokenizer
24
 
25
  model = "meta-llama/Llama-2-7b-chat-hf"
26
  tokenizer = AutoTokenizer.from_pretrained(model, token=hf_token)
27
 
28
+ # tokenizer = AutoTokenizer.from_pretrained(model, token=hf_token)
29
+
30
  # pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
31
 
32
  from transformers import pipeline