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

2024-07-13 10:42:11 -04:00 · 2024-07-13 10:42:11 -04:00 · 19ec5ea3d2
parent 258d6c1244
commit 19ec5ea3d2
1 changed files with 1 additions and 0 deletions
--- a/scripts/hook.py
+++ b/scripts/hook.py
@ -620,6 +620,7 @@ class UnetHook(nn.Module):
                        [
                            t.int_value()
                            for t in param.union_control_types
+                            if t != ControlNetUnionControlType.UNKNOWN
                        ]
                        if param.control_model_type == ControlModelType.ControlNetUnion
                        else None