Spaces:
Runtime error
Runtime error
Commit
Β·
693a1eb
1
Parent(s):
a51a924
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
torch==1.12.1
|
2 |
torchvision==0.13.1
|
|
|
3 |
salesforce-lavis
|
4 |
bitsandbytes
|
5 |
git+https://github.com/huggingface/transformers.git
|
|
|
1 |
torch==1.12.1
|
2 |
torchvision==0.13.1
|
3 |
+
cudatoolkit
|
4 |
salesforce-lavis
|
5 |
bitsandbytes
|
6 |
git+https://github.com/huggingface/transformers.git
|