shobhitjethani commited on
Commit
3df5cc4
1 Parent(s): f152dfa

Update llama_chat_bot.py

Browse files
Files changed (1) hide show
  1. llama_chat_bot.py +3 -3
llama_chat_bot.py CHANGED
@@ -7,8 +7,8 @@ Original file is located at
7
  https://colab.research.google.com/drive/1BBWPVOOR0790ZALdTJy_pYV5bFf8KaGp
8
  """
9
  # curl --proto "=https" --tlsv1.2 -sSf https://sh.rustup.rs | sh
10
- pip install -e transformers torch accelerate
11
- pip install -e --upgrade gradio
12
 
13
 
14
  from transformers import AutoTokenizer
@@ -16,7 +16,7 @@ from transformers import AutoTokenizer
16
  model = "meta-llama/Llama-2-7b-chat-hf"
17
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
18
 
19
- pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
20
 
21
  from transformers import pipeline
22
 
 
7
  https://colab.research.google.com/drive/1BBWPVOOR0790ZALdTJy_pYV5bFf8KaGp
8
  """
9
  # curl --proto "=https" --tlsv1.2 -sSf https://sh.rustup.rs | sh
10
+ # pip install -e transformers torch accelerate
11
+ # pip install -e --upgrade gradio
12
 
13
 
14
  from transformers import AutoTokenizer
 
16
  model = "meta-llama/Llama-2-7b-chat-hf"
17
  tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
18
 
19
+ # pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
20
 
21
  from transformers import pipeline
22