wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Revert category update of dataset

Browse Source
This commit is contained in:
Alexis Rolland 2026-05-14 11:02:29 +08:00
parent 3cbc51861b
commit 36c02d5e9a
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
comfy_extras/nodes_dataset.py
View File

@ -48,7 +48,7 @@ class LoadImageDataSetFromFolderNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="LoadImageDataSetFromFolder", node_id="LoadImageDataSetFromFolder",
display_name="Load Image Dataset from Folder", display_name="Load Image Dataset from Folder",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
io.Combo.Input( io.Combo.Input(
@ -85,7 +85,7 @@ class LoadImageTextDataSetFromFolderNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="LoadImageTextDataSetFromFolder", node_id="LoadImageTextDataSetFromFolder",
display_name="Load Image and Text Dataset from Folder", display_name="Load Image and Text Dataset from Folder",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
io.Combo.Input( io.Combo.Input(
@ -207,7 +207,7 @@ class SaveImageDataSetToFolderNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="SaveImageDataSetToFolder", node_id="SaveImageDataSetToFolder",
display_name="Save Image Dataset to Folder", display_name="Save Image Dataset to Folder",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_output_node=True, is_output_node=True,
is_input_list=True, # Receive images as list is_input_list=True, # Receive images as list
@ -247,7 +247,7 @@ class SaveImageTextDataSetToFolderNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="SaveImageTextDataSetToFolder", node_id="SaveImageTextDataSetToFolder",
display_name="Save Image and Text Dataset to Folder", display_name="Save Image and Text Dataset to Folder",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_output_node=True, is_output_node=True,
is_input_list=True, # Receive both images and texts as lists is_input_list=True, # Receive both images and texts as lists
@ -403,7 +403,7 @@ class ImageProcessingNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id=cls.node_id, node_id=cls.node_id,
display_name=cls.display_name or cls.node_id, display_name=cls.display_name or cls.node_id,
category="training/dataset/image", category="dataset/image",
is_experimental=True, is_experimental=True,
is_input_list=is_group, # True for group, False for individual is_input_list=is_group, # True for group, False for individual
inputs=inputs, inputs=inputs,
@ -552,7 +552,7 @@ class TextProcessingNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id=cls.node_id, node_id=cls.node_id,
display_name=cls.display_name or cls.node_id, display_name=cls.display_name or cls.node_id,
category="training/dataset/text", category="dataset/text",
is_experimental=True, is_experimental=True,
is_input_list=is_group, # True for group, False for individual is_input_list=is_group, # True for group, False for individual
inputs=inputs, inputs=inputs,
@ -824,7 +824,7 @@ class ShuffleImageTextDatasetNode(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="ShuffleImageTextDataset", node_id="ShuffleImageTextDataset",
display_name="Shuffle Image-Text Dataset", display_name="Shuffle Image-Text Dataset",
category="training/dataset/image", category="dataset/image",
is_experimental=True, is_experimental=True,
is_input_list=True, is_input_list=True,
inputs=[ inputs=[
@ -1143,7 +1143,7 @@ class ResolutionBucket(io.ComfyNode):
return io.Schema( return io.Schema(
node_id="ResolutionBucket", node_id="ResolutionBucket",
display_name="Resolution Bucket", display_name="Resolution Bucket",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_input_list=True, is_input_list=True,
inputs=[ inputs=[
@ -1236,7 +1236,7 @@ class MakeTrainingDataset(io.ComfyNode):
node_id="MakeTrainingDataset", node_id="MakeTrainingDataset",
search_aliases=["encode dataset"], search_aliases=["encode dataset"],
display_name="Make Training Dataset", display_name="Make Training Dataset",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_input_list=True, # images and texts as lists is_input_list=True, # images and texts as lists
inputs=[ inputs=[
@ -1322,7 +1322,7 @@ class SaveTrainingDataset(io.ComfyNode):
node_id="SaveTrainingDataset", node_id="SaveTrainingDataset",
search_aliases=["export training data"], search_aliases=["export training data"],
display_name="Save Training Dataset", display_name="Save Training Dataset",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
is_output_node=True, is_output_node=True,
is_input_list=True, # Receive lists is_input_list=True, # Receive lists
@ -1424,7 +1424,7 @@ class LoadTrainingDataset(io.ComfyNode):
node_id="LoadTrainingDataset", node_id="LoadTrainingDataset",
search_aliases=["import dataset", "training data"], search_aliases=["import dataset", "training data"],
display_name="Load Training Dataset", display_name="Load Training Dataset",
category="training/dataset", category="dataset",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
io.String.Input( io.String.Input(