diff --git a/main.py b/main.py index dd6526e13..85903e690 100644 --- a/main.py +++ b/main.py @@ -237,6 +237,8 @@ if __name__ == "__main__": folder_paths.add_model_folder_path("vae", os.path.join(folder_paths.get_output_directory(), "vae")) folder_paths.add_model_folder_path("diffusion_models", os.path.join(folder_paths.get_output_directory(), "diffusion_models")) folder_paths.add_model_folder_path("loras", os.path.join(folder_paths.get_output_directory(), "loras")) + folder_paths.add_model_folder_path("latents", os.path.join(folder_paths.get_output_directory(), "latents")) + if args.input_directory: input_dir = os.path.abspath(args.input_directory) diff --git a/nodes.py b/nodes.py index df4124544..3ded3c304 100644 --- a/nodes.py +++ b/nodes.py @@ -482,9 +482,8 @@ 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("latents"), {"tooltip": "The name of the latent from output/latents to load."})}} + CATEGORY = "_for_testing"