ayushsaini1207 commited on
Commit
a9bdb6d
1 Parent(s): 0185d95

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -2,6 +2,8 @@ streamlit
2
  transformers
3
  peft
4
  torch --index-url https://download.pytorch.org/whl/cu118
 
 
5
  bitsandbytes
6
  accelerate
7
  -i https://pypi.org/simple/ bitsandbytes
 
2
  transformers
3
  peft
4
  torch --index-url https://download.pytorch.org/whl/cu118
5
+ torchvision --index-url https://download.pytorch.org/whl/cu118
6
+ torchaudio --index-url https://download.pytorch.org/whl/cu118
7
  bitsandbytes
8
  accelerate
9
  -i https://pypi.org/simple/ bitsandbytes