wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 13:50:15 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

fix wrong output_list flag

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-11-18 20:45:05 +08:00
parent 0318d2d60c
commit b90069a695
1 changed files with 48 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
comfy_extras/nodes_dataset.py
View File

@ -329,7 +329,7 @@ class ImageProcessingNode(io.ComfyNode):
description = None description = None
extra_inputs = [] extra_inputs = []
is_group_process = None # None = auto-detect, True/False = explicit is_group_process = None # None = auto-detect, True/False = explicit
is_output_list = True # Configurable output mode is_output_list = None # None = auto-detect based on processing mode
@classmethod @classmethod
def _detect_processing_mode(cls): def _detect_processing_mode(cls):
@ -342,10 +342,26 @@ class ImageProcessingNode(io.ComfyNode):
if cls.is_group_process is not None: if cls.is_group_process is not None:
return cls.is_group_process return cls.is_group_process
# Check which method is overridden # Check which method is overridden by looking at the defining class in MRO
base_class = ImageProcessingNode base_class = ImageProcessingNode
has_process = cls._process is not base_class._process
has_group = cls._group_process is not base_class._group_process # Find which class in MRO defines _process
process_definer = None
for klass in cls.__mro__:
if "_process" in klass.__dict__:
process_definer = klass
break
# Find which class in MRO defines _group_process
group_definer = None
for klass in cls.__mro__:
if "_group_process" in klass.__dict__:
group_definer = klass
break
# Check what was overridden (not defined in base class)
has_process = process_definer is not None and process_definer is not base_class
has_group = group_definer is not None and group_definer is not base_class
if has_process and has_group: if has_process and has_group:
raise ValueError( raise ValueError(
@ -366,6 +382,13 @@ class ImageProcessingNode(io.ComfyNode):
is_group = cls._detect_processing_mode() is_group = cls._detect_processing_mode()
# Auto-detect is_output_list if not explicitly set
# Single processing: True (backend collects results into list)
# Group processing: True by default (can be False for single-output nodes)
output_is_list = (
cls.is_output_list if cls.is_output_list is not None else is_group
)
inputs = [ inputs = [
io.Image.Input( io.Image.Input(
"images", "images",
@ -386,7 +409,7 @@ class ImageProcessingNode(io.ComfyNode):
outputs=[ outputs=[
io.Image.Output( io.Image.Output(
display_name="images", display_name="images",
is_output_list=cls.is_output_list, is_output_list=output_is_list,
tooltip="Processed images", tooltip="Processed images",
) )
], ],
@ -470,7 +493,7 @@ class TextProcessingNode(io.ComfyNode):
description = None description = None
extra_inputs = [] extra_inputs = []
is_group_process = None # None = auto-detect, True/False = explicit is_group_process = None # None = auto-detect, True/False = explicit
is_output_list = True # Configurable output mode is_output_list = None # None = auto-detect based on processing mode
@classmethod @classmethod
def _detect_processing_mode(cls): def _detect_processing_mode(cls):
@ -483,10 +506,26 @@ class TextProcessingNode(io.ComfyNode):
if cls.is_group_process is not None: if cls.is_group_process is not None:
return cls.is_group_process return cls.is_group_process
# Check which method is overridden # Check which method is overridden by looking at the defining class in MRO
base_class = TextProcessingNode base_class = TextProcessingNode
has_process = cls._process is not base_class._process
has_group = cls._group_process is not base_class._group_process # Find which class in MRO defines _process
process_definer = None
for klass in cls.__mro__:
if "_process" in klass.__dict__:
process_definer = klass
break
# Find which class in MRO defines _group_process
group_definer = None
for klass in cls.__mro__:
if "_group_process" in klass.__dict__:
group_definer = klass
break
# Check what was overridden (not defined in base class)
has_process = process_definer is not None and process_definer is not base_class
has_group = group_definer is not None and group_definer is not base_class
if has_process and has_group: if has_process and has_group:
raise ValueError( raise ValueError(