Update requirements.txt
Browse files- requirements.txt +8 -3
requirements.txt
CHANGED
|
@@ -1,5 +1,11 @@
|
|
| 1 |
-
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
opencv-python>=4.9.0.80
|
| 4 |
diffusers==0.31.0
|
| 5 |
transformers>=4.49.0
|
|
@@ -38,4 +44,3 @@ onnxscript
|
|
| 38 |
onnxconverter_common
|
| 39 |
flask
|
| 40 |
flask-socketio
|
| 41 |
-
torchao
|
|
|
|
| 1 |
+
# so we use pytorch 2.6
|
| 2 |
+
torch==2.6.0
|
| 3 |
+
torchvision==0.21.0
|
| 4 |
+
torchao==0.9.0
|
| 5 |
+
|
| 6 |
+
#torch>=2.4.0
|
| 7 |
+
#torchvision>=0.19.0
|
| 8 |
+
|
| 9 |
opencv-python>=4.9.0.80
|
| 10 |
diffusers==0.31.0
|
| 11 |
transformers>=4.49.0
|
|
|
|
| 44 |
onnxconverter_common
|
| 45 |
flask
|
| 46 |
flask-socketio
|
|
|