Vision-CAIR commited on
Commit
673400c
1 Parent(s): 1e594e0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -3
requirements.txt CHANGED
@@ -1,10 +1,8 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch==1.12.1
3
- torchvision==0.13.1
4
  salesforce-lavis
5
  bitsandbytes
6
  accelerate
7
  git+https://github.com/huggingface/transformers.git
8
  Cython==0.29.36
9
- cudatoolkit
10
  spacy
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==2.0.0
 
3
  salesforce-lavis
4
  bitsandbytes
5
  accelerate
6
  git+https://github.com/huggingface/transformers.git
7
  Cython==0.29.36
 
8
  spacy