lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-05-11 00:10:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow draw mask or upload mask

Browse Source
This commit is contained in:
lllyasviel
2024-02-01 16:56:02 -08:00
parent 421070b44e
commit 06759ccd7f
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py
View File

@@ -348,10 +348,18 @@ class ControlNetUiGroup(object):
) as self.mask_image_group:
self.mask_image = gr.Image(
value=None,
label="Upload Mask",
label="Mask",
elem_id=f"{elem_id_tabname}_{tabname}_mask_image",
elem_classes=["cnet-mask-image"],
interactive=True,
brush_radius=20,
type="numpy",
tool="sketch",
brush_color=shared.opts.img2img_inpaint_mask_brush_color
if hasattr(
shared.opts, "img2img_inpaint_mask_brush_color"
)
else None,
)
with gr.Tab(label="Batch") as self.batch_tab:
@@ -459,7 +467,7 @@ class ControlNetUiGroup(object):
visible=not self.is_img2img,
)
self.mask_upload = gr.Checkbox(
label="Mask Upload",
label="Use Mask",
value=False,
elem_classes=["cnet-mask-upload"],
elem_id=f"{elem_id_tabname}_{tabname}_controlnet_mask_upload_checkbox",
@@ -1082,7 +1090,7 @@ class ControlNetUiGroup(object):
# Clear mask_image if unchecked.
(gr.update(visible=False), gr.update(value=None))
if not checked
else (gr.update(visible=True), gr.update())
else (gr.update(visible=True), gr.update(value=np.zeros(shape=(512, 512, 3), dtype=np.uint8)))
),
inputs=[self.mask_upload],
outputs=[self.mask_image_group, self.mask_image],