wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-29 21:13:33 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

remove size and add default mode

Browse Source
This commit is contained in:
Charles Chan 2026-03-29 10:50:38 +08:00
parent 8c6953da59
commit 0120eb6861
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy_extras/nodes_dataset.py
View File

@ -196,8 +196,8 @@ def save_images_to_folder(image_list, output_dir, prefix="image", overwrite=True
if overwrite: if overwrite:
filename = f"{prefix}_{idx:05d}.png" filename = f"{prefix}_{idx:05d}.png"
else: else:
full_output_folder, filename, counter, subfolder, prefix = folder_paths.get_save_image_path(prefix, output_dir, img_tensor[0].shape[1], img_tensor[0].shape[0]) _, _, counter, _, resolved_prefix = folder_paths.get_save_image_path(prefix, output_dir)
filename = f"{prefix}_{counter:05}_{idx:05d}.png" filename = f"{resolved_prefix}_{counter:05}_{idx:05d}.png"
filepath = os.path.join(output_dir, filename) filepath = os.path.join(output_dir, filename)
img.save(filepath) img.save(filepath)
saved_files.append(filename) saved_files.append(filename)
@ -230,7 +230,8 @@ class SaveImageDataSetToFolderNode(io.ComfyNode):
), ),
io.Combo.Input( io.Combo.Input(
"mode", "mode",
options=['overwrite', 'increment'], default="overwrite",
options=["overwrite", "increment"],
tooltip="Whether to overwrite existing files or increment filenames to avoid overwriting." tooltip="Whether to overwrite existing files or increment filenames to avoid overwriting."
), ),
], ],
@ -277,7 +278,8 @@ class SaveImageTextDataSetToFolderNode(io.ComfyNode):
), ),
io.Combo.Input( io.Combo.Input(
"mode", "mode",
options=['overwrite', 'increment'], default="overwrite",
options=["overwrite", "increment"],
tooltip="Whether to overwrite existing files or increment filenames to avoid overwriting." tooltip="Whether to overwrite existing files or increment filenames to avoid overwriting."
), ),
], ],