xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-11 05:57:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added Custom Directory Path for Saver in Nodegraph

Browse Source
This commit is contained in:
r42-chun 2024-09-02 03:54:05 +01:00
parent bc7174a6ae
commit 05ab36030b
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -1474,7 +1474,8 @@ class SaveImage:
return { return {
"required": { "required": {
"images": ("IMAGE", {"tooltip": "The images to save."}), "images": ("IMAGE", {"tooltip": "The images to save."}),
"filename_prefix": ("STRING", {"default": "ComfyUI", "tooltip": "The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."}) "filename_prefix": ("STRING", {"default": "ComfyUI", "tooltip": "The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."}),
"output_dir": ("STRING", {"default": "%USERPROFILE%\Documents\ComfyUIOutput", "tooltip": "A folder you want to save the images to instead"})
}, },
"hidden": { "hidden": {
"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO" "prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"
@ -1493,6 +1494,12 @@ class SaveImage:
# =========================== # ===========================
if not output_dir: if not output_dir:
output_dir = self.output_dir output_dir = self.output_dir
output_dir = os.path.expandvars(output_dir)
if not os.path.isdir(output_dir):
logging.error(f"Output directory: {output_dir} does not exist")
raise FileNotFoundError(f"Output directory: {output_dir} does not exist")
return { "ui": { "images": () } }
# =========================== # ===========================
filename_prefix += self.prefix_append filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, output_dir, images[0].shape[1], images[0].shape[0]) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, output_dir, images[0].shape[1], images[0].shape[0])