diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index af8b8d0e..ecb06e8e 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -338,6 +338,7 @@ class ControlNetForForgeOfficial(scripts.Script): preprocessor_output = preprocessor( input_image=input_image, + input_mask=input_mask, resolution=unit.processor_res, slider_1=unit.threshold_a, slider_2=unit.threshold_b, diff --git a/modules_forge/supported_preprocessor.py b/modules_forge/supported_preprocessor.py index 7de23d6a..a6440aa5 100644 --- a/modules_forge/supported_preprocessor.py +++ b/modules_forge/supported_preprocessor.py @@ -65,7 +65,7 @@ class Preprocessor: def process_after_every_sampling(self, process, params, *args, **kwargs): return - def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs): + def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, input_mask=None, **kwargs): return input_image