feat: make cnet_stack optional

This commit is contained in:
philhk
2023-09-25 23:27:30 +02:00
parent ca3f5ae870
commit 9a49da2c19

View File

@@ -358,8 +358,8 @@ class TSC_Apply_ControlNet_Stack:
@classmethod
def INPUT_TYPES(cls):
return {"required": {"positive": ("CONDITIONING",),
"negative": ("CONDITIONING",),
"cnet_stack": ("CONTROL_NET_STACK",)},
"negative": ("CONDITIONING",)},
"optional": {"cnet_stack": ("CONTROL_NET_STACK",)}
}
RETURN_TYPES = ("CONDITIONING","CONDITIONING",)
@@ -367,7 +367,10 @@ class TSC_Apply_ControlNet_Stack:
FUNCTION = "apply_cnet_stack"
CATEGORY = "Efficiency Nodes/Stackers"
def apply_cnet_stack(self, positive, negative, cnet_stack):
def apply_cnet_stack(self, positive, negative, cnet_stack=None):
if cnet_stack is None:
return (positive, negative)
for control_net_tuple in cnet_stack:
control_net, image, strength, start_percent, end_percent = control_net_tuple
controlnet_conditioning = ControlNetApplyAdvanced().apply_controlnet(positive, negative,