Spaces:
Sleeping
Sleeping
ayushsaini1207
commited on
Commit
•
a9bdb6d
1
Parent(s):
0185d95
Update requirements.txt
Browse files- 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
|