Allow skipping of control type for ControlNet union (#2995)

Mikubill · Jul 13, 2024 · 19ec5ea · 19ec5ea
1 parent 258d6c1
commit 19ec5ea
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/hook.py b/scripts/hook.py
@@ -620,6 +620,7 @@ def forward(self, x, timesteps=None, context=None, y=None, **kwargs):
                         [
                             t.int_value()
                             for t in param.union_control_types
+                            if t != ControlNetUnionControlType.UNKNOWN
                         ]
                         if param.control_model_type == ControlModelType.ControlNetUnion
                         else None