Merge remote-tracking branch 'origin'
Browse files- requirements.txt +11 -3
requirements.txt
CHANGED
|
@@ -1,10 +1,19 @@
|
|
| 1 |
-
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
opencv-python>=4.9.0.80
|
| 4 |
diffusers==0.31.0
|
| 5 |
transformers>=4.49.0
|
| 6 |
tokenizers>=0.20.3
|
| 7 |
accelerate>=1.1.1
|
|
|
|
|
|
|
|
|
|
| 8 |
tqdm
|
| 9 |
imageio
|
| 10 |
easydict
|
|
@@ -35,4 +44,3 @@ onnxscript
|
|
| 35 |
onnxconverter_common
|
| 36 |
flask
|
| 37 |
flask-socketio
|
| 38 |
-
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
|
| 12 |
tokenizers>=0.20.3
|
| 13 |
accelerate>=1.1.1
|
| 14 |
+
|
| 15 |
+
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 16 |
+
|
| 17 |
tqdm
|
| 18 |
imageio
|
| 19 |
easydict
|
|
|
|
| 44 |
onnxconverter_common
|
| 45 |
flask
|
| 46 |
flask-socketio
|
|
|