From 19ec5ea3d21193eefe7161f162552b3d52649d20 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Sat, 13 Jul 2024 10:42:11 -0400
Subject: [PATCH] Allow skipping of control type for ControlNet union (#2995)

---
 scripts/hook.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/hook.py b/scripts/hook.py
index 4e485e375..a01ffd477 100644
--- 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