From 597fe3e049f8c1f13867cf754d6fe57d296ae5f7 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 1 Feb 2024 21:38:48 -0800
Subject: [PATCH] Update forge_ipadapter.py

---
 .../sd_forge_ipadapter/scripts/forge_ipadapter.py             | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py b/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
index b4167604..47b72044 100644
--- a/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
+++ b/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
@@ -25,7 +25,6 @@ class PreprocessorClipVisionForIPAdapter(PreprocessorClipVision):
             weight_type="original",
             noise=0.0,
             embeds=None,
-            attn_mask=None,
             unfold_batch=False,
         )
         return cond
@@ -49,7 +48,6 @@ class PreprocessorClipVisionWithInsightFaceForIPAdapter(PreprocessorClipVisionFo
             weight_type="original",
             noise=0.0,
             embeds=None,
-            attn_mask=None,
             unfold_batch=False,
         )
         return cond
@@ -81,7 +79,6 @@ class PreprocessorInsightFaceForInstantID(Preprocessor):
             weight_type="original",
             noise=0.0,
             embeds=None,
-            attn_mask=None,
             unfold_batch=False,
             instant_id=True
         )
@@ -155,6 +152,7 @@ class IPAdapterPatcher(ControlModelPatcher):
             end_at=self.end_percent,
             faceid_v2=self.faceid_v2,
             weight_v2=self.weight_v2,
+            attn_mask=mask.squeeze(1) if mask is not None else None,
             **cond,
         )[0]