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

Add outputs' id/name

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-11-11 16:23:24 +08:00
parent b85ed81792
commit 1f852778ec
2 changed files with 42 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
comfy_extras/nodes_dataset.py
View File

@ -59,7 +59,9 @@ class LoadImageDataSetFromFolderNode(io.ComfyNode):
) )
], ],
outputs=[ outputs=[
io.Image.Output(is_output_list=True, tooltip="List of loaded images") io.Image.Output(
"images", is_output_list=True, tooltip="List of loaded images"
)
], ],
) )
@ -92,8 +94,12 @@ class LoadImageTextDataSetFromFolderNode(io.ComfyNode):
) )
], ],
outputs=[ outputs=[
io.Image.Output(is_output_list=True, tooltip="List of loaded images"), io.Image.Output(
io.String.Output(is_output_list=True, tooltip="List of text captions"), "images", is_output_list=True, tooltip="List of loaded images"
),
io.String.Output(
"texts", is_output_list=True, tooltip="List of text captions"
),
], ],
) )
@ -329,7 +335,11 @@ class ImageProcessingNode(io.ComfyNode):
is_experimental=True, is_experimental=True,
is_input_list=True, is_input_list=True,
inputs=inputs, inputs=inputs,
outputs=[io.Image.Output(is_output_list=True, tooltip="Processed images")], outputs=[
io.Image.Output(
"images", is_output_list=True, tooltip="Processed images"
)
],
) )
@classmethod @classmethod
@ -393,7 +403,11 @@ class TextProcessingNode(io.ComfyNode):
is_experimental=True, is_experimental=True,
is_input_list=True, is_input_list=True,
inputs=inputs, inputs=inputs,
outputs=[io.String.Output(is_output_list=True, tooltip="Processed texts")], outputs=[
io.String.Output(
"texts", is_output_list=True, tooltip="Processed texts"
)
],
) )
@classmethod @classmethod
@ -752,8 +766,12 @@ class ShuffleImageTextDatasetNode(io.ComfyNode):
), ),
], ],
outputs=[ outputs=[
io.Image.Output(is_output_list=True, tooltip="Shuffled images"), io.Image.Output(
io.String.Output(is_output_list=True, tooltip="Shuffled texts"), "images", is_output_list=True, tooltip="Shuffled images"
),
io.String.Output(
"texts", is_output_list=True, tooltip="Shuffled texts"
),
], ],
) )
@ -884,9 +902,13 @@ class MakeTrainingDataset(io.ComfyNode):
), ),
], ],
outputs=[ outputs=[
io.Latent.Output(is_output_list=True, tooltip="List of latent dicts"), io.Latent.Output(
"latents", is_output_list=True, tooltip="List of latent dicts"
),
io.Conditioning.Output( io.Conditioning.Output(
is_output_list=True, tooltip="List of conditioning lists" "conditioning",
is_output_list=True,
tooltip="List of conditioning lists",
), ),
], ],
) )
@ -1059,9 +1081,13 @@ class LoadTrainingDataset(io.ComfyNode):
), ),
], ],
outputs=[ outputs=[
io.Latent.Output(is_output_list=True, tooltip="List of latent dicts"), io.Latent.Output(
"latents", is_output_list=True, tooltip="List of latent dicts"
),
io.Conditioning.Output( io.Conditioning.Output(
is_output_list=True, tooltip="List of conditioning lists" "conditioning",
is_output_list=True,
tooltip="List of conditioning lists",
), ),
], ],
) )

10
comfy_extras/nodes_train.py
View File

@ -387,10 +387,10 @@ class TrainLoraNode(io.ComfyNode):
), ),
], ],
outputs=[ outputs=[
io.Model.Output(tooltip="Model with LoRA applied"), io.Model.Output("model", tooltip="Model with LoRA applied"),
io.Custom("LORA_MODEL").Output(tooltip="LoRA weights"), io.Custom("LORA_MODEL").Output("lora", tooltip="LoRA weights"),
io.Custom("LOSS_MAP").Output(tooltip="Loss history"), io.Custom("LOSS_MAP").Output("loss_map", tooltip="Loss history"),
io.Int.Output(tooltip="Total training steps"), io.Int.Output("steps", tooltip="Total training steps"),
], ],
) )
@ -663,7 +663,7 @@ class LoraModelLoader(io.ComfyNode):
), ),
], ],
outputs=[ outputs=[
io.Model.Output(tooltip="The modified diffusion model."), io.Model.Output("model", tooltip="The modified diffusion model."),
], ],
) )