Switch to blocks

app_df.py

@@ -27,7 +27,7 @@ def generate_diffusion_forced_video(
     shift=8.0,
     inference_steps=30,
     use_usp=False,
-    offload=False,
+    offload=True,
     fps=24,
     seed=None,
     prompt_enhancer=False,
@@ -128,32 +128,74 @@ def generate_diffusion_forced_video(
 resolution_options = ["540P", "720P"]
 model_options = ["Skywork/SkyReels-V2-DF-1.3B-540P"]  # Update if there are more
 
-gr.Interface(
-    fn=generate_diffusion_forced_video,
-    inputs=[
-        gr.Textbox(label="Prompt"),
-        gr.Dropdown(choices=model_options, value=model_options[0], label="Model ID"),
-        gr.Radio(choices=resolution_options, value="540P", label="Resolution", interactive=False),
-        gr.Slider(minimum=16, maximum=200, value=97, step=1, label="Number of Frames"),
-        gr.Image(type="filepath", label="Input Image (optional)"),
-        gr.Number(label="AR Step", value=0),
-        gr.Checkbox(label="Causal Attention"),
-        gr.Number(label="Causal Block Size", value=1),
-        gr.Number(label="Base Num Frames", value=97),
-        gr.Number(label="Overlap History (set for long videos)", value=None),
-        gr.Number(label="AddNoise Condition", value=0),
-        gr.Slider(minimum=1.0, maximum=20.0, value=6.0, step=0.1, label="Guidance Scale"),
-        gr.Slider(minimum=0.0, maximum=20.0, value=8.0, step=0.1, label="Shift"),
-        gr.Slider(minimum=1, maximum=100, value=30, step=1, label="Inference Steps"),
-        gr.Checkbox(label="Use USP"),
-        gr.Checkbox(label="Offload", value=True, interactive=False),
-        gr.Slider(minimum=1, maximum=60, value=24, step=1, label="FPS"),
-        gr.Number(label="Seed (optional)", precision=0),
-        gr.Checkbox(label="Prompt Enhancer"),
-        gr.Checkbox(label="Use TeaCache"),
-        gr.Slider(minimum=0.0, maximum=1.0, value=0.2, step=0.01, label="TeaCache Threshold"),
-        gr.Checkbox(label="Use Retention Steps"),
-    ],
-    outputs=gr.Video(label="Generated Video"),
-    title="SkyReels V2 Diffusion Forcing"
-).launch(show_error=True, show_api=False, share=False)
+with gr.Blocks() as demo:
+    with gr.Column():
+        gr.Markdown("# SkyReels V2")
+
+        with gr.Row():
+
+            with gr.Column():
+
+                prompt = gr.Textbox(label="Prompt")
+                model_id = gr.Dropdown(choices=model_options, value=model_options[0], label="Model ID")
+                resolution = gr.Radio(choices=resolution_options, value="540P", label="Resolution", interactive=False)
+                num_frames = gr.Slider(minimum=16, maximum=200, value=97, step=1, label="Number of Frames")
+                image = gr.Image(type="filepath", label="Input Image (optional)")
+                
+                with gr.Accordion("Advanced Settings", open=False):
+                    ar_step = gr.Number(label="AR Step", value=0)
+                    causal_attention = gr.Checkbox(label="Causal Attention")
+                    causal_block_size = gr.Number(label="Causal Block Size", value=1)
+                    base_num_frames = gr.Number(label="Base Num Frames", value=97)
+                    overlap_history = gr.Number(label="Overlap History (set for long videos)", value=None)
+                    addnoise_condition = gr.Number(label="AddNoise Condition", value=0)
+                    guidance_scale = gr.Slider(minimum=1.0, maximum=20.0, value=6.0, step=0.1, label="Guidance Scale")
+                    shift = gr.Slider(minimum=0.0, maximum=20.0, value=8.0, step=0.1, label="Shift")
+                    inference_steps = gr.Slider(minimum=1, maximum=100, value=30, step=1, label="Inference Steps")
+                    use_usp = gr.Checkbox(label="Use USP")
+                    offload = gr.Checkbox(label="Offload", value=True, interactive=False)
+                    fps = gr.Slider(minimum=1, maximum=60, value=24, step=1, label="FPS")
+                    seed = gr.Number(label="Seed (optional)", precision=0)
+                    prompt_enhancer = gr.Checkbox(label="Prompt Enhancer")
+                    use_teacache = gr.Checkbox(label="Use TeaCache")
+                    teacache_thresh = gr.Slider(minimum=0.0, maximum=1.0, value=0.2, step=0.01, label="TeaCache Threshold")
+                    use_ret_steps = gr.Checkbox(label="Use Retention Steps")
+
+                submit_btn = gr.Button("Generate")
+
+            with gr.Column():
+
+                output_video = gr.Video(label="Generated Video")
+
+    submit_btn.click(
+        fn = generate_diffusion_forced_video,
+        inputs = [
+            prompt,
+            model_id,
+            resolution,
+            num_frames,
+            image,
+            ar_step,
+            causal_attention,
+            causal_block_size,
+            base_num_frames,
+            overlap_history,
+            addnoise_condition,
+            guidance_scale,
+            shift,
+            inference_steps,
+            use_usp,
+            offload,
+            fps,
+            seed,
+            prompt_enhancer,
+            teacache,
+            teacache_thresh,
+            use_ret_steps
+        ],
+        outputs = [
+            output_video
+        ]
+    )
+
+demo.launch(show_error=True, show_api=False, share=False)