DiningSystem commited on
Commit
d38a8e1
1 Parent(s): a62cd60

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu116
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
 
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-cuda113
6
  accelerate
7
  git+https://github.com/huggingface/transformers.git