Update app.py

app.py

@@ -40,14 +40,16 @@ def pipeline_debug_info(pipe):
     info = []
     info.append("=== PIPELINE DEBUG INFO ===")
 
-    # Transformer info
     try:
         tr = pipe.transformer.config
         info.append(f"Transformer Class: {pipe.transformer.__class__.__name__}")
-        info.append(f"Hidden size: {tr.get('hidden_size')}")
-        info.append(f"Attention heads: {tr.get('num_attention_heads')}")
-        info.append(f"Layers: {tr.get('num_layers')}")
+
+        # Z-Image-Turbo correct keys
+        info.append(f"Hidden dim: {tr.get('hidden_dim')}")
+        info.append(f"Attention heads: {tr.get('num_heads')}")
+        info.append(f"Depth (layers): {tr.get('depth')}")
         info.append(f"Patch size: {tr.get('patch_size')}")
+        info.append(f"MLP ratio: {tr.get('mlp_ratio')}")
         info.append(f"Attention backend: {tr.get('attn_implementation')}")
     except Exception as e:
         info.append(f"Transformer diagnostics failed: {e}")
@@ -62,6 +64,7 @@ def pipeline_debug_info(pipe):
 
     return "\n".join(info)
 
+
 def latent_shape_info(h, w, pipe):
     try:
         c = pipe.vae.config.latent_channels
@@ -141,8 +144,8 @@ examples = [
     ["Portrait of a wise old wizard..."],
 ]
 
-with gr.Blocks(title="Z-Image-Turbo Debug Demo") as demo:
-    gr.Markdown("# 🎨 Z-Image-Turbo — Multi Image + Full Debug Logs")
+with gr.Blocks(title="Z-Image-Turbo Multi Image Demo") as demo:
+    gr.Markdown("# 🎨 Z-Image-Turbo — Multi Image ")
 
     with gr.Row():
         with gr.Column(scale=1):