Update models.py

models.py

@@ -243,7 +243,23 @@ def set_clip_skip(pipe):
     if hasattr(pipe, 'text_encoder'):
         print(f"  [OK] CLIP skip {CLIP_SKIP}")
 
+        
+def load_image_encoder():
+    """Load CLIP Image Encoder for IP-Adapter."""
+    print("Loading CLIP Image Encoder for IP-Adapter...")
+    try:
+        image_encoder = CLIPVisionModelWithProjection.from_pretrained(
+            "h94/IP-Adapter",
+            subfolder="models/image_encoder",
+            torch_dtype=dtype
+        ).to(device)
+        print("  [OK] CLIP Image Encoder loaded successfully")
+        return image_encoder
+    except Exception as e:
+        print(f"  [ERROR] Could not load image encoder: {e}")
+        return None
+
 
-__all__ = ['draw_kps', 'fuse_lora_with_scale']
+__all__ = ['draw_kps', 'fuse_lora_with_scale', 'load_image_encoder']
 
 print("[OK] models.py ready - NO MultiControlNetModel, following examplewithface.py")