298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-12 18:33:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Read input from storage.

Browse Source
This commit is contained in:
Symbiomatrix 2025-04-11 16:24:08 +03:00 committed by GitHub
parent 57bb55baca
commit f52e089d86
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
nodes.py
View File

@ -499,9 +499,10 @@ class SaveLatent:
class LoadLatent:
@classmethod
def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory()
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f)) and f.endswith(".latent")]
return {"required": {"latent": [sorted(files), ]}, }
return {"required":
{"latent": (folder_paths.get_filename_list("latent"), )
},
}
CATEGORY = "_for_testing"
@ -1652,10 +1653,9 @@ class PreviewImage(SaveImage):
class LoadImage:
@classmethod
def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory()
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
return {"required":
{"image": (sorted(files), {"image_upload": True})},
return {"required":
{"image": (folder_paths.get_filename_list("input"), {"image_upload": True})
},
}
CATEGORY = "image"
@ -1728,11 +1728,10 @@ class LoadImageMask:
_color_channels = ["alpha", "red", "green", "blue"]
@classmethod
def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory()
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
return {"required":
{"image": (sorted(files), {"image_upload": True}),
"channel": (s._color_channels, ), }
return {"required":
{"image": (folder_paths.get_filename_list("input"), {"image_upload": True}),
"channel": (s._color_channels, ),
},
}
CATEGORY = "mask"