Update models.py

models.py

@@ -17,7 +17,10 @@ from transformers import CLIPVisionModelWithProjection
 from insightface.app import FaceAnalysis
 from controlnet_aux import ZoeDetector, OpenposeDetector, LeresDetector, MidasDetector, MediapipeFaceDetector
 from huggingface_hub import hf_hub_download, snapshot_download
-from compel import Compel, ReturnedEmbeddingsType
+
+# --- START FIX: Import CompelForSDXL ---
+from compel import CompelForSDXL 
+# --- END FIX ---
 
 # Use reference implementation's attention processor
 from attention_processor import IPAttnProcessor2_0, AttnProcessor
@@ -382,21 +385,19 @@ def setup_ip_adapter(pipe, image_encoder):
         return None, False
 
 
+# --- START FIX: Use CompelForSDXL helper class ---
 def setup_compel(pipe):
-    """Setup Compel for better SDXL prompt handling."""
-    print("Setting up Compel for enhanced prompt processing...")
+    """Setup CompelForSDXL for better SDXL prompt handling."""
+    print("Setting up Compel for enhanced prompt processing (using CompelForSDXL)...")
     try:
-        compel = Compel(
-            tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
-            text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
-            returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
-            requires_pooled=[False, True]
-        )
-        print("  [OK] Compel loaded successfully")
+        # Pass the whole pipeline as per Compel docs
+        compel = CompelForSDXL(pipe) 
+        print("  [OK] CompelForSDXL loaded successfully")
         return compel, True
     except Exception as e:
         print(f"  [WARNING] Compel not available: {e}")
         return None, False
+# --- END FIX ---
 
 
 def setup_scheduler(pipe):
@@ -414,7 +415,7 @@ def optimize_pipeline(pipe):
     # Try to enable xformers
     if device == "cuda":
         try:
-            pipe.enable_xformfiers_memory_efficient_attention()
+            pipe.enable_xformers_memory_efficient_attention()
             print("  [OK] xformers enabled")
         except Exception as e:
             print(f"  [INFO] xformers not available: {e}")