Vision-CAIR commited on
Commit
a51a924
1 Parent(s): 2139551

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch
2
- torchvision
3
  salesforce-lavis
4
  bitsandbytes
5
  git+https://github.com/huggingface/transformers.git
 
1
+ torch==1.12.1
2
+ torchvision==0.13.1
3
  salesforce-lavis
4
  bitsandbytes
5
  git+https://github.com/huggingface/transformers.git