Wan-AI

app.py

@@ -125,7 +125,14 @@ def get_duration(image,
                  seed,
                  progress):
     """Calculate dynamic GPU duration based on parameters."""
-    return sampling_steps * 15
+    if duration_seconds >= 3:
+        return 220
+    elif sampling_steps > 35 and duration_seconds >= 2:
+        return 180
+    elif sampling_steps < 35 or duration_seconds < 2:
+        return 105
+    else:
+        return 90
 
 # --- 2. Gradio Inference Function ---
 @spaces.GPU(duration=get_duration)

requirements.txt

@@ -11,6 +11,5 @@ easydict
 ftfy
 dashscope
 imageio-ffmpeg
-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
+https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.8/flash_attn-2.7.4.post1+cu126torch2.7-cp310-cp310-linux_x86_64.whl
 numpy>=1.23.5,<2
-