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: add controlnet union model

Browse Source
This commit is contained in:
Dowon
2024-07-17 20:27:56 +09:00
parent ab845cf596
commit db3be81c4e
3 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
controlnet_ext/common.py
View File

@@ -8,4 +8,5 @@ cn_model_module = {
"tile": "tile_resample",
"depth": "depth_midas",
}
cn_model_regex = re.compile("|".join(cn_model_module.keys()), flags=re.IGNORECASE)
_names = [*cn_model_module, "union"]
cn_model_regex = re.compile("|".join(_names), flags=re.IGNORECASE)

6
controlnet_ext/controlnet_ext.py
View File

@@ -61,13 +61,13 @@ class ControlNetExt:
if (not self.cn_available) or model == "None":
return
if module is None or module == "None":
if module == "None":
module = None
if module is None:
for m, v in cn_model_module.items():
if m in model:
module = v
break
else:
module = None
cn_units = [
self.external_cn.ControlNetUnit(