Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
6d0f162
1
Parent(s):
5760e26
Lora name change
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
|
|
| 20 |
|
| 21 |
# Additional enhancement LoRAs for FusionX-like quality
|
| 22 |
ACCVIDEO_LORA_REPO = "alibaba-pai/Wan2.1-Fun-Reward-LoRAs"
|
| 23 |
-
MPS_LORA_FILENAME = "Wan2.1-
|
| 24 |
|
| 25 |
# Load enhanced model components
|
| 26 |
print("🚀 Loading FusionX Enhanced Wan2.1 I2V Model...")
|
|
@@ -330,11 +330,11 @@ def get_duration(input_image, prompt, height, width,
|
|
| 330 |
progress):
|
| 331 |
# FusionX optimized duration calculation
|
| 332 |
if steps > 8 and duration_seconds > 3:
|
| 333 |
-
return
|
| 334 |
elif steps > 8 or duration_seconds > 3:
|
| 335 |
-
return
|
| 336 |
else:
|
| 337 |
-
return
|
| 338 |
|
| 339 |
@spaces.GPU(duration=get_duration)
|
| 340 |
def generate_video(input_image, prompt, height, width,
|
|
|
|
| 20 |
|
| 21 |
# Additional enhancement LoRAs for FusionX-like quality
|
| 22 |
ACCVIDEO_LORA_REPO = "alibaba-pai/Wan2.1-Fun-Reward-LoRAs"
|
| 23 |
+
MPS_LORA_FILENAME = "Wan2.1-Fun-14B-InP-MPS.safetensors"
|
| 24 |
|
| 25 |
# Load enhanced model components
|
| 26 |
print("🚀 Loading FusionX Enhanced Wan2.1 I2V Model...")
|
|
|
|
| 330 |
progress):
|
| 331 |
# FusionX optimized duration calculation
|
| 332 |
if steps > 8 and duration_seconds > 3:
|
| 333 |
+
return 600
|
| 334 |
elif steps > 8 or duration_seconds > 3:
|
| 335 |
+
return 300
|
| 336 |
else:
|
| 337 |
+
return 150
|
| 338 |
|
| 339 |
@spaces.GPU(duration=get_duration)
|
| 340 |
def generate_video(input_image, prompt, height, width,
|