caohy666 commited on
Commit
d5629cd
·
1 Parent(s): 910b840

<fix> move flash_attn installation to app.py.

Browse files
Files changed (2) hide show
  1. app.py +4 -1
  2. 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 --no-build-isolation
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