feature: partial add of inpainting controlnet

todo: figure out how to input "mask" for this mode
This commit is contained in:
Bryce 2023-05-05 01:27:20 -07:00 committed by Bryce Drennan
parent 9526a85a71
commit b5356b143f
2 changed files with 8 additions and 0 deletions

View File

@ -213,6 +213,13 @@ CONTROLNET_CONFIGS = [
weights_url="https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/69fc48b9cbd98661f6d0288dc59b59a5ccb32a6b/control_v11e_sd15_ip2p.pth",
alias="edit",
),
ControlNetConfig(
short_name="inpaint15",
control_type="inpaint",
config_path="configs/control-net-v15.yaml",
weights_url="https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/69fc48b9cbd98661f6d0288dc59b59a5ccb32a6b/control_v11p_sd15_inpaint.pth",
alias="inpaint",
),
]
CONTROLNET_CONFIG_SHORTCUTS = {m.short_name: m for m in CONTROLNET_CONFIGS}

View File

@ -198,4 +198,5 @@ CONTROL_MODES = {
# "scribble": None,
"shuffle": shuffle_map_torch,
"edit": noop,
"inpaint": noop,
}