🔧 feat: Support depth_hand_refiner ControlNet preprocessor (#460)

This commit is contained in:
Chenlei Hu
2024-01-04 10:52:16 +00:00
committed by GitHub
parent f1dfb34ed1
commit 31dda6f0bc
3 changed files with 4 additions and 2 deletions

View File

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

View File

@@ -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"],
}

View File

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