feat: add controlnet union model

2026-01-26 19:29:54 +00:00 · 2024-07-17 20:27:56 +09:00
parent ab845cf596
commit db3be81c4e
3 changed files with 9 additions and 4 deletions
--- a/controlnet_ext/controlnet_ext.py
+++ b/controlnet_ext/controlnet_ext.py
@@ -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(