bowmanchow commited on
Commit
1bff495
·
1 Parent(s): 0346ed8

change torch version

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch>=2.4.0
2
- torchvision>=0.19.0
3
  opencv-python>=4.9.0.80
4
  diffusers==0.31.0
5
  transformers>=4.49.0
@@ -37,7 +37,7 @@ dominate
37
  # onnxconverter_common
38
  flask
39
  flask-socketio
40
- torchao
41
  hydra-core
42
  git+https://github.com/facebookresearch/segment-anything.git
43
  kornia
 
1
+ torch>=2.4.0,<2.8.0
2
+ torchvision>=0.19.0,<0.23.0
3
  opencv-python>=4.9.0.80
4
  diffusers==0.31.0
5
  transformers>=4.49.0
 
37
  # onnxconverter_common
38
  flask
39
  flask-socketio
40
+ torchao<0.9.0
41
  hydra-core
42
  git+https://github.com/facebookresearch/segment-anything.git
43
  kornia