diff --git a/adetailer/args.py b/adetailer/args.py index 0331af7..f205347 100644 --- a/adetailer/args.py +++ b/adetailer/args.py @@ -18,8 +18,8 @@ from pydantic import ( validator, ) -cn_model_regex = r".*(inpaint|tile|scribble|lineart|openpose).*|^None$" -cn_module_regex = r".*(inpaint|tile|pidi|lineart|openpose).*|^None$" +cn_model_regex = r".*(inpaint|tile|scribble|lineart|openpose|depth).*|^None$" +cn_module_regex = r".*(inpaint|tile|pidi|lineart|openpose|depth).*|^None$" @dataclass diff --git a/adetailer/ui.py b/adetailer/ui.py index e49ea5e..fa28262 100644 --- a/adetailer/ui.py +++ b/adetailer/ui.py @@ -26,6 +26,7 @@ cn_module_choices = { "openpose": ["openpose_full", "dw_openpose_full"], "tile": ["tile_resample", "tile_colorfix", "tile_colorfix+sharp"], "scribble": ["t2ia_sketch_pidi"], + "depth": ["depth_midas", "depth_hand_refiner"], } diff --git a/controlnet_ext/controlnet_ext.py b/controlnet_ext/controlnet_ext.py index 37c5a71..9f54f12 100644 --- a/controlnet_ext/controlnet_ext.py +++ b/controlnet_ext/controlnet_ext.py @@ -48,6 +48,7 @@ cn_model_module = { "lineart": "lineart_coarse", "openpose": "openpose_full", "tile": "tile_resample", + "depth": "depth_midas", } cn_model_regex = re.compile("|".join(cn_model_module.keys()))