Update models.py

models.py

@@ -385,14 +385,14 @@ def setup_ip_adapter(pipe, image_encoder):
         return None, False
 
 
-# --- START FIX: Use CompelForSDXL helper class ---
+# --- START FIX: Use CompelForSDXL and ENABLE TRUNCATION ---
 def setup_compel(pipe):
     """Setup CompelForSDXL for better SDXL prompt handling."""
     print("Setting up Compel for enhanced prompt processing (using CompelForSDXL)...")
     try:
-        # Pass the whole pipeline as per Compel docs
-        compel = CompelForSDXL(pipe) 
-        print("  [OK] CompelForSDXL loaded successfully")
+        # Pass the whole pipeline and enable truncation
+        compel = CompelForSDXL(pipe, truncate_long_prompts=True) 
+        print("  [OK] CompelForSDXL loaded successfully (with truncation)")
         return compel, True
     except Exception as e:
         print(f"  [WARNING] Compel not available: {e}")