set examples caching

app_df.py

@@ -20,10 +20,11 @@ if is_shared_ui:
 
 def generate_diffusion_forced_video(
     prompt,
-    model_id,
-    resolution,
-    num_frames,
     image=None,
+    target_length="4",
+    model_id="Skywork/SkyReels-V2-DF-1.3B-540P",
+    resolution="540P",
+    num_frames=97,
     ar_step=0,
     causal_attention=False,
     causal_block_size=1,
@@ -211,12 +212,16 @@ with gr.Blocks() as demo:
 
                 gr.Examples(
                     examples = [
-                        ["A graceful white swan with a curved neck and delicate feathers swimming in a serene lake at dawn, its reflection perfectly mirrored in the still water as mist rises from the surface, with the swan occasionally dipping its head into the water to feed.", "./examples/swan.jpeg"],
+                        ["A graceful white swan with a curved neck and delicate feathers swimming in a serene lake at dawn, its reflection perfectly mirrored in the still water as mist rises from the surface, with the swan occasionally dipping its head into the water to feed.", "./examples/swan.jpeg", "10"],
                        # ["A graceful white swan with a curved neck and delicate feathers swimming in a serene lake at dawn, its reflection perfectly mirrored in the still water as mist rises from the surface, with the swan occasionally dipping its head into the water to feed.", None],
-                        ["A sea turtle swimming near a shipwreck", "./examples/turtle.jpeg"],
+                        ["A sea turtle swimming near a shipwreck", "./examples/turtle.jpeg", "10"],
                        # ["A sea turtle swimming near a shipwreck", None],
                     ],
-                    inputs = [prompt, image]
+                    fn = generate_diffusion_forced_video,
+                    inputs = [prompt, image, target_length],
+                    outputs = [output_video],
+                    cache_examples = True,
+                    cache_mode = "lazy"
                 )
 
     def set_num_frames(target_l):
@@ -285,10 +290,11 @@ with gr.Blocks() as demo:
         fn = generate_diffusion_forced_video,
         inputs = [
             prompt,
+            image,
+            target_length,
             model_id,
             resolution,
             num_frames,
-            image,
             ar_step,
             causal_attention,
             causal_block_size,