Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 19:29:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: update controlnet preprocessors

Browse Source
This commit is contained in:
Dowon
2023-10-19 22:41:43 +09:00
parent ec6c993ec6
commit dc0f789b07
2 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
controlnet_ext/controlnet_ext.py
View File

@@ -29,7 +29,7 @@ cn_model_module = {
"scribble": "t2ia_sketch_pidi",
"lineart": "lineart_coarse",
"openpose": "openpose_full",
"tile": None,
"tile": "tile_resample",
}
cn_model_regex = re.compile("|".join(cn_model_module.keys()))
@@ -60,7 +60,7 @@ class ControlNetExt:
if (not self.cn_available) or model == "None":
return
if module is None:
if module is None or module == "None":
for m, v in cn_model_module.items():
if m in model:
module = v