Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +4 -5
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
torch==2.
|
| 2 |
-
torchvision==0.
|
| 3 |
opencv-python>=4.9.0.80
|
| 4 |
diffusers>=0.31.0
|
| 5 |
transformers>=4.49.0
|
|
@@ -11,6 +11,5 @@ easydict
|
|
| 11 |
ftfy
|
| 12 |
dashscope
|
| 13 |
imageio-ffmpeg
|
| 14 |
-
|
| 15 |
-
numpy>=1.23.5,<2
|
| 16 |
-
|
|
|
|
| 1 |
+
torch==2.8.0
|
| 2 |
+
torchvision==0.23.0
|
| 3 |
opencv-python>=4.9.0.80
|
| 4 |
diffusers>=0.31.0
|
| 5 |
transformers>=4.49.0
|
|
|
|
| 11 |
ftfy
|
| 12 |
dashscope
|
| 13 |
imageio-ffmpeg
|
| 14 |
+
https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.18/flash_attn-2.7.4+cu128torch2.8-cp310-cp310-linux_x86_64.whl
|
| 15 |
+
numpy>=1.23.5,<2
|
|
|