Update app.py

app.py

@@ -36,25 +36,25 @@ pipe.to("cuda")
 # ------------------------
 # ATTENTION + PIPE INFO
 # ------------------------
-def pipeline_debug_info():
+def pipeline_debug_info(pipe):
     info = []
-    info.append("=== PIPELINE DEBUG INFO ===")
-    info.append(f"UNet attention backend: {pipe.unet.config.attn_implementation}")
-    info.append(f"Transformer attention backend: {pipe.transformer.config.attn_implementation}")
-    
-    # Processor classes
+
+    # Transformer config
     try:
-        info.append(f"UNet mid-block processor: {type(pipe.unet.mid_block.attentions[0].processor)}")
+        info.append(f"Transformer attention backend: {pipe.transformer.config.attn_implementation}")
     except:
-        info.append("UNet mid-block processor: <not found>")
+        info.append("No transformer.attn_implementation")
 
+    # Processor class
     try:
-        info.append(f"Transformer block processor: {type(pipe.transformer.blocks[0].attn.processor)}")
-    except:
-        info.append("Transformer block processor: <not found>")
+        proc = pipe.transformer.blocks[0].attn.processor
+        info.append(f"Processor type: {type(proc)}")
+    except Exception as e:
+        info.append(f"Processor error: {e}")
 
     return "\n".join(info)
 
+
 # ------------------------
 # IMAGE GENERATOR
 # ------------------------