wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-12 23:30:56 +08:00
Code Issues Actions 7 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=[
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=[
io.Image.Output(is_output_list=True, tooltip="List of loaded images"),
io.String.Output(is_output_list=True, tooltip="List of text captions"),
io.Image.Output(
"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_input_list=True,
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
@ -393,7 +403,11 @@ class TextProcessingNode(io.ComfyNode):
is_experimental=True,
is_input_list=True,
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
@ -752,8 +766,12 @@ class ShuffleImageTextDatasetNode(io.ComfyNode):
),
],
outputs=[
io.Image.Output(is_output_list=True, tooltip="Shuffled images"),
io.String.Output(is_output_list=True, tooltip="Shuffled texts"),
io.Image.Output(
"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=[
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(
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=[
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(
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=[
io.Model.Output(tooltip="Model with LoRA applied"),
io.Custom("LORA_MODEL").Output(tooltip="LoRA weights"),
io.Custom("LOSS_MAP").Output(tooltip="Loss history"),
io.Int.Output(tooltip="Total training steps"),
io.Model.Output("model", tooltip="Model with LoRA applied"),
io.Custom("LORA_MODEL").Output("lora", tooltip="LoRA weights"),
io.Custom("LOSS_MAP").Output("loss_map", tooltip="Loss history"),
io.Int.Output("steps", tooltip="Total training steps"),
],
)
@ -663,7 +663,7 @@ class LoraModelLoader(io.ComfyNode):
),
],
outputs=[
io.Model.Output(tooltip="The modified diffusion model."),
io.Model.Output("model", tooltip="The modified diffusion model."),
],
)