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 bad id/naming

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

44
comfy_extras/nodes_dataset.py
View File

@ -48,7 +48,7 @@ class LoadImageDataSetFromFolderNode(io.ComfyNode):
def define_schema(cls): def define_schema(cls):
return io.Schema( return io.Schema(
node_id="LoadImageDataSetFromFolder", node_id="LoadImageDataSetFromFolder",
display_name="Load Simple Image Dataset from Folder", display_name="Load Image Dataset from Folder",
category="dataset", category="dataset",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
@ -60,7 +60,9 @@ class LoadImageDataSetFromFolderNode(io.ComfyNode):
], ],
outputs=[ outputs=[
io.Image.Output( io.Image.Output(
"images", is_output_list=True, tooltip="List of loaded images" display_name="images",
is_output_list=True,
tooltip="List of loaded images",
) )
], ],
) )
@ -83,7 +85,7 @@ class LoadImageTextDataSetFromFolderNode(io.ComfyNode):
def define_schema(cls): def define_schema(cls):
return io.Schema( return io.Schema(
node_id="LoadImageTextDataSetFromFolder", node_id="LoadImageTextDataSetFromFolder",
display_name="Load Simple Image and Text Dataset from Folder", display_name="Load Image and Text Dataset from Folder",
category="dataset", category="dataset",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
@ -95,10 +97,14 @@ class LoadImageTextDataSetFromFolderNode(io.ComfyNode):
], ],
outputs=[ outputs=[
io.Image.Output( io.Image.Output(
"images", is_output_list=True, tooltip="List of loaded images" display_name="images",
is_output_list=True,
tooltip="List of loaded images",
), ),
io.String.Output( io.String.Output(
"texts", is_output_list=True, tooltip="List of text captions" display_name="texts",
is_output_list=True,
tooltip="List of text captions",
), ),
], ],
) )
@ -201,7 +207,7 @@ class SaveImageDataSetToFolderNode(io.ComfyNode):
def define_schema(cls): def define_schema(cls):
return io.Schema( return io.Schema(
node_id="SaveImageDataSetToFolder", node_id="SaveImageDataSetToFolder",
display_name="Save Simple Image Dataset to Folder", display_name="Save Image Dataset to Folder",
category="dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_output_node=True, is_output_node=True,
@ -240,7 +246,7 @@ class SaveImageTextDataSetToFolderNode(io.ComfyNode):
def define_schema(cls): def define_schema(cls):
return io.Schema( return io.Schema(
node_id="SaveImageTextDataSetToFolder", node_id="SaveImageTextDataSetToFolder",
display_name="Save Simple Image and Text Dataset to Folder", display_name="Save Image and Text Dataset to Folder",
category="dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_output_node=True, is_output_node=True,
@ -337,7 +343,9 @@ class ImageProcessingNode(io.ComfyNode):
inputs=inputs, inputs=inputs,
outputs=[ outputs=[
io.Image.Output( io.Image.Output(
"images", is_output_list=True, tooltip="Processed images" display_name="images",
is_output_list=True,
tooltip="Processed images",
) )
], ],
) )
@ -405,7 +413,7 @@ class TextProcessingNode(io.ComfyNode):
inputs=inputs, inputs=inputs,
outputs=[ outputs=[
io.String.Output( io.String.Output(
"texts", is_output_list=True, tooltip="Processed texts" display_name="texts", is_output_list=True, tooltip="Processed texts"
) )
], ],
) )
@ -767,10 +775,12 @@ class ShuffleImageTextDatasetNode(io.ComfyNode):
], ],
outputs=[ outputs=[
io.Image.Output( io.Image.Output(
"images", is_output_list=True, tooltip="Shuffled images" display_name="images",
is_output_list=True,
tooltip="Shuffled images",
), ),
io.String.Output( io.String.Output(
"texts", is_output_list=True, tooltip="Shuffled texts" display_name="texts", is_output_list=True, tooltip="Shuffled texts"
), ),
], ],
) )
@ -903,10 +913,12 @@ class MakeTrainingDataset(io.ComfyNode):
], ],
outputs=[ outputs=[
io.Latent.Output( io.Latent.Output(
"latents", is_output_list=True, tooltip="List of latent dicts" display_name="latents",
is_output_list=True,
tooltip="List of latent dicts",
), ),
io.Conditioning.Output( io.Conditioning.Output(
"conditioning", display_name="conditioning",
is_output_list=True, is_output_list=True,
tooltip="List of conditioning lists", tooltip="List of conditioning lists",
), ),
@ -1082,10 +1094,12 @@ class LoadTrainingDataset(io.ComfyNode):
], ],
outputs=[ outputs=[
io.Latent.Output( io.Latent.Output(
"latents", is_output_list=True, tooltip="List of latent dicts" display_name="latents",
is_output_list=True,
tooltip="List of latent dicts",
), ),
io.Conditioning.Output( io.Conditioning.Output(
"conditioning", display_name="conditioning",
is_output_list=True, is_output_list=True,
tooltip="List of conditioning lists", tooltip="List of conditioning lists",
), ),

18
comfy_extras/nodes_train.py
View File

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