alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: make cnet_stack optional

Browse Source
This commit is contained in:
philhk
2023-09-25 23:27:30 +02:00
parent ca3f5ae870
commit 9a49da2c19
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
efficiency_nodes.py
View File

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