Cleanup

app.py

@@ -24,8 +24,9 @@ LANDSCAPE_HEIGHT = 480
 MAX_SEED = np.iinfo(np.int32).max
 
 FIXED_FPS = 24
-MIN_FRAMES_MODEL = 8
-MAX_FRAMES_MODEL = 81
+MIN_FRAMES_MODEL = 13
+MAX_FRAMES_MODEL = 121
+NUM_FRAMES_DEFAULT = 81
 
 
 pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
@@ -89,15 +90,18 @@ def get_duration(
     randomize_seed,
     progress,
 ):
-    return steps * 15
+    forward_duration_base = 8
+    forward_duration = forward_duration_base * (num_frames / NUM_FRAMES_DEFAULT)**1.5
+    forward_count = 2 if guidance_scale > 1 else 1
+    return 10 + steps * forward_count * forward_duration
 
 @spaces.GPU(duration=get_duration)
 def generate_video(
     input_image,
     prompt,
     negative_prompt=default_negative_prompt,
-    num_frames = MAX_FRAMES_MODEL,
-    guidance_scale = 3.5,
+    num_frames = NUM_FRAMES_DEFAULT,
+    guidance_scale = 1,
     steps = 28,
     seed = 42,
     randomize_seed = False,
@@ -172,14 +176,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             input_image_component = gr.Image(type="pil", label="Input Image (auto-resized to target H/W)")
             prompt_input = gr.Textbox(label="Prompt", value=default_prompt_i2v)
-            num_frames_input = gr.Slider(minimum=MIN_FRAMES_MODEL, maximum=MAX_FRAMES_MODEL, step=1, value=MAX_FRAMES_MODEL, label="Frames")
+            num_frames_input = gr.Slider(minimum=MIN_FRAMES_MODEL, maximum=MAX_FRAMES_MODEL, step=1, value=NUM_FRAMES_DEFAULT, label="Frames")
             
             with gr.Accordion("Advanced Settings", open=False):
                 negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
                 seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
                 randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
                 steps_slider = gr.Slider(minimum=1, maximum=40, step=1, value=28, label="Inference Steps")
-                guidance_scale_input = gr.Slider(minimum=0.0, maximum=20.0, step=0.5, value=1.0, label="Guidance Scale")
+                guidance_scale_input = gr.Slider(minimum=1.0, maximum=20.0, step=0.5, value=1.0, label="Guidance Scale")
 
             generate_button = gr.Button("Generate Video", variant="primary")
         with gr.Column():