File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed
Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change 11UNION_CONTROLNET_TYPES = {
2- "auto" : - 1 ,
32 "openpose" : 0 ,
43 "depth" : 1 ,
54 "hed/pidi/scribble/ted" : 2 ,
Original file line number Diff line number Diff line change @@ -4,7 +4,7 @@ class SetUnionControlNetType:
44 @classmethod
55 def INPUT_TYPES (s ):
66 return {"required" : {"control_net" : ("CONTROL_NET" , ),
7- "type" : (list (UNION_CONTROLNET_TYPES .keys ()),)
7+ "type" : ([ "auto" ] + list (UNION_CONTROLNET_TYPES .keys ()),)
88 }}
99
1010 CATEGORY = "conditioning/controlnet"
@@ -14,7 +14,7 @@ def INPUT_TYPES(s):
1414
1515 def set_controlnet_type (self , control_net , type ):
1616 control_net = control_net .copy ()
17- type_number = UNION_CONTROLNET_TYPES [ type ]
17+ type_number = UNION_CONTROLNET_TYPES . get ( type , - 1 )
1818 if type_number >= 0 :
1919 control_net .set_extra_arg ("control_type" , [type_number ])
2020 else :
You can’t perform that action at this time.
0 commit comments