Timing logs; drop reference example

app.py

@@ -8,6 +8,7 @@ import spaces  # must be imported before torch / CUDA-touching libs
 import random
 import re
 import tempfile
+import time
 from types import SimpleNamespace
 
 import einops
@@ -269,6 +270,7 @@ def edit_image(
     Returns:
         The edited image and the seed that was used.
     """
+    t0 = time.time()
     if source_image is None:
         raise gr.Error("Please upload an image to edit.")
     if not instruction or not instruction.strip():
@@ -305,8 +307,10 @@ def edit_image(
     )
     if task == "ti_i2i":
         kwargs["reference_image"] = reference_image.convert("RGB")
+    t1 = time.time()
     with torch.no_grad():
         out = pipe(**kwargs)
+    print(f"[timing] edit_image: prep {t1 - t0:.1f}s, pipe {time.time() - t1:.1f}s")
     return to_pil_first_frame(out.videos), seed
 
 
@@ -341,6 +345,7 @@ def edit_video(
     """
     from decord import VideoReader
 
+    t0 = time.time()
     if not source_video:
         raise gr.Error("Please upload a video to edit.")
     if not instruction or not instruction.strip():
@@ -392,8 +397,10 @@ def edit_video(
         ref_path = tempfile.NamedTemporaryFile(suffix=".png", delete=False).name
         reference_image.convert("RGB").save(ref_path)
         kwargs["reference_image"] = ref_path
+    t1 = time.time()
     with torch.no_grad():
         out = pipe(**kwargs)
+    print(f"[timing] edit_video: prep {t1 - t0:.1f}s, pipe {time.time() - t1:.1f}s ({num_frames} frames)")
 
     video = out.videos
     if video.ndim == 5:
@@ -466,15 +473,6 @@ with gr.Blocks() as demo:
                     cache_mode="lazy",
                     label="Image examples",
                 )
-                gr.Examples(
-                    examples=[["examples/dog.jpg", "change the dog to a [V*]", "examples/red_fox.jpg"]],
-                    inputs=[img_src, img_prompt, img_ref],
-                    outputs=[img_out, img_seed],
-                    fn=edit_image,
-                    cache_examples=True,
-                    cache_mode="lazy",
-                    label="Reference-guided example",
-                )
 
             # ---------------- Video editing ----------------
             with gr.Tab("Video editing"):