whyumesh commited on
Commit
4f25957
·
verified ·
1 Parent(s): 4142d39

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,8 +1,10 @@
1
- torch
 
 
2
  transformers
3
  Pillow
4
  opencv-python
5
  numpy
6
  av
7
  qwen_vl_utils
8
- torchvision
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.1.0+cu118
3
+ torchvision==0.16.0+cu118
4
  transformers
5
  Pillow
6
  opencv-python
7
  numpy
8
  av
9
  qwen_vl_utils
10
+ accelerate>=0.26.0