Spaces:
Running
on
Zero
Running
on
Zero
<fix> move flash_attn installation to app.py.
Browse files- app.py +4 -1
- requirements.txt +1 -1
app.py
CHANGED
@@ -1,4 +1,8 @@
|
|
1 |
import os
|
|
|
|
|
|
|
|
|
2 |
import sys
|
3 |
import torch
|
4 |
import diffusers
|
@@ -18,7 +22,6 @@ from PIL import Image, ImageDraw, ImageFilter
|
|
18 |
from models import HunyuanVideoTransformer3DModel
|
19 |
from pipelines import HunyuanVideoImageToVideoPipeline
|
20 |
|
21 |
-
|
22 |
header = """
|
23 |
# DRA-Ctrl Gradio App
|
24 |
|
|
|
1 |
import os
|
2 |
+
|
3 |
+
if 'SPACES_APP' in os.environ:
|
4 |
+
os.system("pip install flash-attn==2.7.3 --no-build-isolation")
|
5 |
+
|
6 |
import sys
|
7 |
import torch
|
8 |
import diffusers
|
|
|
22 |
from models import HunyuanVideoTransformer3DModel
|
23 |
from pipelines import HunyuanVideoImageToVideoPipeline
|
24 |
|
|
|
25 |
header = """
|
26 |
# DRA-Ctrl Gradio App
|
27 |
|
requirements.txt
CHANGED
@@ -2,7 +2,7 @@ torch==2.4.1
|
|
2 |
torchvision==0.19.1
|
3 |
diffusers==0.33.1
|
4 |
transformers==4.45.0
|
5 |
-
flash-attn==2.7.3
|
6 |
gradio
|
7 |
omegaconf
|
8 |
peft
|
|
|
2 |
torchvision==0.19.1
|
3 |
diffusers==0.33.1
|
4 |
transformers==4.45.0
|
5 |
+
# flash-attn==2.7.3
|
6 |
gradio
|
7 |
omegaconf
|
8 |
peft
|