Hugging Face's logo

Spaces:
Vision-CAIR
/
minigpt4
Running on A10G

App Files Community
35
Vision-CAIR commited on
Commit
693a1eb
•
1 Parent(s): a51a924

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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