Layout: prompt + run inline above image; advanced settings in accordion below; match slider max_height to height

Bump preview + slider heights to 720/900px

Stream PiD's 4 internal student-sampler steps (yield after each); 'Upscaling with PiD — step K/4' label

Drop intermediate captures + gallery; use gr.ImageSlider for Z-Image vs PiD A/B; dynamic 'Generating Z-Image step X/N' / 'Upscaling' labels

Stream taef1 previews from Z-Image (thread+queue) into a flash gr.Image; swap to gr.Gallery only when all PiD steps done

Fix _latent_to_pil squeeze dim (T=1, dim=1 not dim=0); stream gallery as each PiD step completes

Patch Gemma2.forward to put torch.tensor on the embedding's device (ZeroGPU rejects the CPU normalizer cross-device multiply)

Replace transformers' vmap-based mask broadcaster with explicit broadcasting (ZeroGPU __torch_function__ can't fake-alloc inside vmap)

Force eager attention on Z-Image text encoder (vmap conflict with ZeroGPU); add Citrus theme + dark-mode color css

Initial PiD + Z-Image step-by-step denoising demo for ZeroGPU