Update models.py

models.py

@@ -152,7 +152,7 @@ def load_controlnets():
     """Load ControlNet models."""
     print("Loading ControlNet Zoe Depth model...")
     controlnet_depth = ControlNetModel.from_pretrained(
-        "diffusers/controlnet-zoe-depth-sdxl-1.0",
+        "thibaud/controlnet-openpose-sdxl-1.0",
         torch_dtype=dtype
     ).to(device)
     print("  [OK] ControlNet Depth loaded")
@@ -161,7 +161,7 @@ def load_controlnets():
     print("Loading ControlNet OpenPose model...")
     try:
         controlnet_openpose = ControlNetModel.from_pretrained(
-            "diffusers/controlnet-openpose-sdxl-1.0",
+            "thibaud/controlnet-openpose-sdxl-1.0",
             torch_dtype=dtype
         ).to(device)
         print("  [OK] ControlNet OpenPose loaded")
@@ -399,6 +399,14 @@ def setup_scheduler(pipe):
 def optimize_pipeline(pipe):
     """Apply optimizations to pipeline."""
     
+    # Try to enable xformers
+    if device == "cuda":
+        try:
+            pipe.enable_xformers_memory_efficient_attention()
+            print("  [OK] xformers enabled")
+        except Exception as e:
+            print(f"  [INFO] xformers not available: {e}")
+
     # Enable CPU offloading for VRAM-constrained environments
     print("  [OK] Enabling model CPU offloading...")
     pipe.enable_model_cpu_offload()