Update app.py

app.py

@@ -59,6 +59,12 @@ def get_pipe(model_id: str, lora_scale: float = 1.0):
             LORA_MODEL_ID, weight_name="merged_lora_weights.safetensors"
         )
         pipe.fuse_lora(lora_scale=lora_scale)
+        # После загрузки LoRA
+        print("Loaded LoRA weights:")
+        for name, param in pipe.unet.named_parameters():
+            if "lora" in name.lower() and param.requires_grad:
+                print(f"LoRA layer: {name}, shape: {param.shape}")
+                break
     else:
         # Load a standard model without LoRA
         pipe = DiffusionPipeline.from_pretrained(