Vision-CAIR
commited on
Commit
·
6219393
1
Parent(s):
6947959
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
torch
|
2 |
torchvision
|
3 |
salesforce-lavis
|
|
|
4 |
transformers==4.28.1
|
5 |
accelerate==0.16.0
|
6 |
sentence-transformers==2.2.2
|
7 |
-
sentencepiece==0.1.97
|
|
|
1 |
torch
|
2 |
torchvision
|
3 |
salesforce-lavis
|
4 |
+
bitsandbytes
|
5 |
transformers==4.28.1
|
6 |
accelerate==0.16.0
|
7 |
sentence-transformers==2.2.2
|
8 |
+
sentencepiece==0.1.97
|