Function zimage_step 

pub fn zimage_step(a: &mut ZStepArgs<'_>) -> bool

One full DiT forward on the device: x_embed → pad rows → noise refiner ×2 → concat [img, cap] → 30 layers → final LayerNorm·scale → Linear → image rows into a.out.