wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

add CATEGORY for execution nodes

Browse Source
This commit is contained in:
Lt.Dr.Data 2023-07-18 12:37:50 +09:00
parent 8159c6ce37
commit 932dbf0c16
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_execution_control.py
View File

@ -11,6 +11,7 @@ class LoopControl:
RETURN_TYPES = ("*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, **kwargs):
if 'loopback_input' not in kwargs or kwargs['loopback_input'] is None:
current = kwargs['initial_input']
@ -53,6 +54,7 @@ class LoopCounterCondition:
RETURN_TYPES = ("LOOP_CONDITION", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, count, trigger):
return (CounterCondition(count), )
@ -70,6 +72,7 @@ class InputZip:
RETURN_TYPES = ("*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, input1, input2):
return ((input1, input2), )
@ -85,6 +88,7 @@ class InputUnzip:
RETURN_TYPES = ("*", "*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, zipped_input):
input1, input2 = zipped_input
return (input1, input2, )
@ -102,6 +106,7 @@ class ExecutionBlocker:
RETURN_TYPES = ("*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, input, signal):
return input
@ -124,6 +129,7 @@ class ExecutionOneOf:
RETURN_TYPES = ("*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, **kwargs):
if 'input1' in kwargs and kwargs['input1'] is not None:
return (kwargs['input1'], )
@ -157,6 +163,8 @@ class ExecutionSwitch:
RETURN_TYPES = ("*", "*", "*", "*", "*", )
FUNCTION = "doit"
CATEGORY = "execution"
def doit(s, select, input1, input2_opt=None, input3_opt=None, input4_opt=None, input5_opt=None):
if select == 1:
return input1, None, None, None, None