lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-07 00:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

add mask api

Browse Source
This commit is contained in:
lllyasviel
2024-02-01 17:32:13 -08:00
parent 05605ed605
commit d4d794d09e
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -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,

2
modules_forge/supported_preprocessor.py
View File

@@ -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