wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-21 20:00:17 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

make minimal changes for upstream

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-03-26 22:10:25 +09:00
parent a5b8a856ef
commit d726ee6f8e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -726,7 +726,7 @@ class SaveImage:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": return {"required":
{"images": ("IMAGE", ), {"images": ("IMAGE", ),
"filename_prefix": ("STRING", {"default": "ComfyUI"}) }, "filename_prefix": ("STRING", {"default": "ComfyUI"})},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
} }
@ -809,7 +809,9 @@ class LoadImage:
def INPUT_TYPES(s): def INPUT_TYPES(s):
if not os.path.exists(s.input_dir): if not os.path.exists(s.input_dir):
os.makedirs(s.input_dir) os.makedirs(s.input_dir)
return {"required": {"image": (sorted(os.listdir(s.input_dir)), ) }} return {"required":
{"image": (sorted(os.listdir(s.input_dir)), )},
}
WIDGET_TYPES = {"recv img": (["disable", "enable"], )} WIDGET_TYPES = {"recv img": (["disable", "enable"], )}
@ -844,7 +846,8 @@ class LoadImageMask:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": return {"required":
{"image": (sorted(os.listdir(s.input_dir)), ), {"image": (sorted(os.listdir(s.input_dir)), ),
"channel": (["alpha", "red", "green", "blue"], ),}} "channel": (["alpha", "red", "green", "blue"], ),}
}
WIDGET_TYPES = { "recv img": (["disable", "enable"], ) } WIDGET_TYPES = { "recv img": (["disable", "enable"], ) }