Gradio 6 theme placement, tight duration calibration, length-scaled camera speed

README.md

@@ -46,6 +46,17 @@ python -m miniworld.sample --dataset re10k \
   ray-encoding features (`freq=15`, unnormalized translations), so no
   ground-truth poses or reference video are needed.
 * Streaming causal VAE decode-on-commit, matching the reference pipeline.
+* The **Camera speed** slider is `--trajectory_magnitude` expressed in
+  "magnitude at `total_len=64`" units. The trajectory builder spreads the whole
+  path evenly over the rollout, so the authors advise scaling magnitude linearly
+  with length to keep the apparent speed constant (3.0 @ 64 → 4.5 @ 96); the
+  Space does that for you, and at 64 latent frames it is the reference value
+  verbatim.
+
+Measured on this Space's ZeroGPU slice: 46 s at 20 latent frames (77 output
+frames), 72 s at 32 (125 frames), 157 s at 64 (253 frames). The
+`@spaces.GPU(duration=...)` estimate replays the sampler's asynchronous step
+schedule to cost each rollout, so short clips reserve proportionally less quota.
 
 The action-conditioned DROID checkpoint is intentionally not exposed: its
 conditioning requires per-dataset `q01/q99` action normalization statistics that

app.py

@@ -275,10 +275,13 @@ def _rollout_tflops(total_len: int, steps: int) -> float:
     return tflops
 
 
-# Calibrated against measured wall-clock on ZeroGPU (see README).
-_TFLOPS_PER_SEC = 45.0
-_VAE_SEC_PER_LATENT_FRAME = 0.15
-_FIXED_OVERHEAD_SEC = 20.0
+# Calibrated on this Space's ZeroGPU H200 slice: measured 46.0s / 72.3s / 157.2s
+# at total_len 20 / 32 / 64 against 2395 / 3599 / 7703 modelled TFLOP, i.e. a
+# very clean 47.8 TFLOP/s (streaming VAE decode overlaps the denoiser, so it
+# needs no separate term).
+_TFLOPS_PER_SEC = 47.8
+_VAE_SEC_PER_LATENT_FRAME = 0.0
+_FIXED_OVERHEAD_SEC = 2.0
 
 
 def _duration(*args, **kwargs) -> int:
@@ -326,13 +329,20 @@ def simulate(
     denoiser.cfg_scale = float(cfg_scale)
     denoiser.steps = int(num_sampling_steps)
 
+    # `build_custom_trajectory` spreads the whole path evenly over the rollout,
+    # so a fixed magnitude means *faster* per-frame motion in a shorter clip.
+    # The authors' guidance is to scale it linearly with length to keep the
+    # apparent speed constant (3.0 @ total_len 64 -> 4.5 @ 96), so the slider is
+    # exposed as a speed in "magnitude at 64 latent frames" units.
+    magnitude_eff = float(magnitude) * total_len / 64.0
+
     videos = init_frame.unsqueeze(0).unsqueeze(0).to(device)  # (1, 1, H, W, C)
     poses = (
         build_custom_trajectory(
             trajectory,
             num_frames=4 * (total_len - 1) + 1,
             focal_norm=float(focal_norm),
-            magnitude=float(magnitude),
+            magnitude=magnitude_eff,
         )
         .unsqueeze(0)
         .to(device)
@@ -388,7 +398,8 @@ def simulate(
         path,
         seed,
         f"**{n} frames** @ {SAVE_FPS} fps ({n / SAVE_FPS:.1f}s) · "
-        f"{total_len} latent frames · `{trajectory}` @ {magnitude:g} · "
+        f"{total_len} latent frames · `{trajectory}` · speed {magnitude:g} "
+        f"(magnitude {magnitude_eff:.2f}) · "
         f"seed `{seed}` · {elapsed:.1f}s of GPU time",
     )
 
@@ -398,7 +409,7 @@ def simulate(
 # --------------------------------------------------------------------------- #
 CSS = "#col-container { max-width: 1060px; margin: 0 auto; }"
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
+with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             f"""
@@ -432,13 +443,15 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
                     value="orbit_right",
                 )
                 magnitude = gr.Slider(
-                    label="Motion strength",
+                    label="Camera speed",
                     minimum=0.5,
                     maximum=8.0,
                     step=0.5,
                     value=3.0,
-                    info="3.0 gives clear, stable motion over a 32-frame rollout. "
-                    "Scale up for longer rollouts; lower it if late frames smear.",
+                    info="3.0 is the paper's default: clear, stable motion. "
+                    "1.0 is nearly static, 8.0 breaks down late. Scaled "
+                    "internally with rollout length so the apparent speed "
+                    "stays constant.",
                 )
                 total_len = gr.Slider(
                     label="Rollout length (latent frames)",
@@ -519,4 +532,4 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
         outputs=[result, seed, info],
     )
 
-demo.queue().launch(mcp_server=True)
+demo.queue().launch(theme=gr.themes.Citrus(), css=CSS, mcp_server=True)