Update app_quant_latent.py

app_quant_latent.py

@@ -585,13 +585,13 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
                 latent_gallery.append(latent_img)
 
                 # Yield intermediate update: final gallery empty for now
-                yield None, latent_gallery, final_gallery, LOGS
+                yield latent_gallery, None, LOGS  # latent updates, final image empty
 
             # decode final image after all timesteps
             final_img = pipe.decode_latents(latents)[0]
             final_gallery.append(final_img)
             LOGS.append("✅ Advanced latent pipeline succeeded.")
-            yield final_img, latent_gallery, final_gallery, LOGS
+            yield latent_gallery, final_img, LOGS 
 
         except Exception as e:
             LOGS.append(f"⚠️ Advanced latent mode failed: {e}")
@@ -943,7 +943,7 @@ with gr.Blocks(title="Z-Image-Turbo") as demo:
     run_btn.click(
         generate_image,
         inputs=[prompt, height, width, steps, seed],
-        outputs=[final_image, latent_gallery, logs_box]
+        outputs=[latent_gallery, final_image, logs_box] 
     )
 
 demo.launch()