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

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):
return io.Schema(
node_id="LoadImageDataSetFromFolder",
display_name="Load Simple Image Dataset from Folder",
display_name="Load Image Dataset from Folder",
category="dataset",
is_experimental=True,
inputs=[
@ -60,7 +60,9 @@ class LoadImageDataSetFromFolderNode(io.ComfyNode):
],
outputs=[
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):
return io.Schema(
node_id="LoadImageTextDataSetFromFolder",
display_name="Load Simple Image and Text Dataset from Folder",
display_name="Load Image and Text Dataset from Folder",
category="dataset",
is_experimental=True,
inputs=[
@ -95,10 +97,14 @@ class LoadImageTextDataSetFromFolderNode(io.ComfyNode):
],
outputs=[
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(
"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):
return io.Schema(
node_id="SaveImageDataSetToFolder",
display_name="Save Simple Image Dataset to Folder",
display_name="Save Image Dataset to Folder",
category="dataset",
is_experimental=True,
is_output_node=True,
@ -240,7 +246,7 @@ class SaveImageTextDataSetToFolderNode(io.ComfyNode):
def define_schema(cls):
return io.Schema(
node_id="SaveImageTextDataSetToFolder",
display_name="Save Simple Image and Text Dataset to Folder",
display_name="Save Image and Text Dataset to Folder",
category="dataset",
is_experimental=True,
is_output_node=True,
@ -337,7 +343,9 @@ class ImageProcessingNode(io.ComfyNode):
inputs=inputs,
outputs=[
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,
outputs=[
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=[
io.Image.Output(
"images", is_output_list=True, tooltip="Shuffled images"
display_name="images",
is_output_list=True,
tooltip="Shuffled images",
),
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=[
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(
"conditioning",
display_name="conditioning",
is_output_list=True,
tooltip="List of conditioning lists",
),
@ -1082,10 +1094,12 @@ class LoadTrainingDataset(io.ComfyNode):
],
outputs=[
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(
"conditioning",
display_name="conditioning",
is_output_list=True,
tooltip="List of conditioning lists",
),

18
comfy_extras/nodes_train.py
View File

@ -387,10 +387,16 @@ class TrainLoraNode(io.ComfyNode):
),
],
outputs=[
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"),
io.Model.Output(
display_name="model", tooltip="Model with LoRA applied"
),
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=[
io.Model.Output("model", tooltip="The modified diffusion model."),
io.Model.Output(
display_name="model", tooltip="The modified diffusion model."
),
],
)