Spaces:
Running
on
L40S
Running
on
L40S
Update requirements.txt
Browse files- requirements.txt +10 -1
requirements.txt
CHANGED
|
@@ -1,6 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
opencv-python>=4.9.0.80
|
| 2 |
diffusers>=0.31.0
|
| 3 |
transformers>=4.49.0
|
|
|
|
|
|
|
| 4 |
tokenizers>=0.20.3
|
| 5 |
accelerate>=1.1.1
|
| 6 |
tqdm
|
|
@@ -12,4 +17,8 @@ imageio-ffmpeg
|
|
| 12 |
gradio>=5.0.0
|
| 13 |
numpy>=1.23.5,<2
|
| 14 |
xfuser>=0.4.1
|
| 15 |
-
pyloudnorm
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch==2.4.1
|
| 2 |
+
torchvision==0.19.1
|
| 3 |
+
torchaudio==2.4.1
|
| 4 |
opencv-python>=4.9.0.80
|
| 5 |
diffusers>=0.31.0
|
| 6 |
transformers>=4.49.0
|
| 7 |
+
peft
|
| 8 |
+
xformers==0.0.28
|
| 9 |
tokenizers>=0.20.3
|
| 10 |
accelerate>=1.1.1
|
| 11 |
tqdm
|
|
|
|
| 17 |
gradio>=5.0.0
|
| 18 |
numpy>=1.23.5,<2
|
| 19 |
xfuser>=0.4.1
|
| 20 |
+
pyloudnorm
|
| 21 |
+
ninja
|
| 22 |
+
psutil
|
| 23 |
+
packaging
|
| 24 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|