diff --git a/extensions-builtin/sd_forge_controlllite/scripts/forge_controllllite.py b/extensions-builtin/sd_forge_controlllite/scripts/forge_controllllite.py
index 3f347264..4a83a915 100644
--- a/extensions-builtin/sd_forge_controlllite/scripts/forge_controllllite.py
+++ b/extensions-builtin/sd_forge_controlllite/scripts/forge_controllllite.py
@@ -18,7 +18,7 @@ class ControlLLLitePatcher(ControlModelPatcher):
         self.state_dict = state_dict
         return
 
-    def process_before_every_sampling(self, process, cond, *args, **kwargs):
+    def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
         unet = process.sd_model.forge_objects.unet
 
         unet = opLLLiteLoader(
diff --git a/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py b/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
index e1d724b3..b4167604 100644
--- a/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
+++ b/extensions-builtin/sd_forge_ipadapter/scripts/forge_ipadapter.py
@@ -144,7 +144,7 @@ class IPAdapterPatcher(ControlModelPatcher):
         self.weight_v2 = False
         return
 
-    def process_before_every_sampling(self, process, cond, *args, **kwargs):
+    def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
         unet = process.sd_model.forge_objects.unet
 
         unet = opIPAdapterApply(
diff --git a/modules_forge/supported_controlnet.py b/modules_forge/supported_controlnet.py
index 8881488d..17e39040 100644
--- a/modules_forge/supported_controlnet.py
+++ b/modules_forge/supported_controlnet.py
@@ -26,7 +26,7 @@ class ControlModelPatcher:
     def process_after_running_preprocessors(self, process, params, *args, **kwargs):
         return
 
-    def process_before_every_sampling(self, process, cond, *args, **kwargs):
+    def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
         return
 
     def process_after_every_sampling(self, process, params, *args, **kwargs):
@@ -142,7 +142,7 @@ class ControlNetPatcher(ControlModelPatcher):
     def __init__(self, model_patcher):
         super().__init__(model_patcher)
 
-    def process_before_every_sampling(self, process, cond, *args, **kwargs):
+    def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
         unet = process.sd_model.forge_objects.unet
 
         unet = apply_controlnet_advanced(