fix: fix controlnet misc

This commit is contained in:
Dowon
2023-10-20 21:40:25 +09:00
parent 92cee3bbb0
commit 6832136581
2 changed files with 3 additions and 0 deletions

View File

@@ -25,6 +25,7 @@ cn_module_choices = {
],
"openpose": ["openpose_full", "dw_openpose_full"],
"tile": ["tile_resample", "tile_colorfix", "tile_colorfix+sharp"],
"scribble": ["t2ia_sketch_pidi"],
}

View File

@@ -65,6 +65,8 @@ class ControlNetExt:
if m in model:
module = v
break
else:
module = None
cn_units = [
self.external_cn.ControlNetUnit(