thankfulcarp commited on
Commit
6d0f162
·
1 Parent(s): 5760e26

Lora name change

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-MPS-Reward-LoRA.safetensors"
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 100
334
  elif steps > 8 or duration_seconds > 3:
335
- return 80
336
  else:
337
- return 65
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,