"auto" type is only relevant to the SetUnionControlNetType node.

MoonRide303 · Jul 22, 2024 · b2c995f · b2c995f
1 parent 4151fbf
commit b2c995f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/comfy/cldm/control_types.py b/comfy/cldm/control_types.py
@@ -1,5 +1,4 @@
 UNION_CONTROLNET_TYPES = {
-    "auto": -1,
     "openpose": 0,
     "depth": 1,
     "hed/pidi/scribble/ted": 2,

diff --git a/comfy_extras/nodes_controlnet.py b/comfy_extras/nodes_controlnet.py
@@ -4,7 +4,7 @@ class SetUnionControlNetType:
     @classmethod
     def INPUT_TYPES(s):
         return {"required": {"control_net": ("CONTROL_NET", ),
-                             "type": (list(UNION_CONTROLNET_TYPES.keys()),)
+                             "type": (["auto"] + list(UNION_CONTROLNET_TYPES.keys()),)
                              }}
 
     CATEGORY = "conditioning/controlnet"
@@ -14,7 +14,7 @@ def INPUT_TYPES(s):
 
     def set_controlnet_type(self, control_net, type):
         control_net = control_net.copy()
-        type_number = UNION_CONTROLNET_TYPES[type]
+        type_number = UNION_CONTROLNET_TYPES.get(type, -1)
         if type_number >= 0:
             control_net.set_extra_arg("control_type", [type_number])
         else: