From cadadc89cc711618c1555f48bded89338578f92a Mon Sep 17 00:00:00 2001 From: Vijay Date: Sat, 9 Nov 2024 18:38:00 +0530 Subject: [PATCH 1/2] Latent files saved under Input/Latent but require manual move to Input for loading --- main.py | 2 ++ nodes.py | 7 +++++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index c23210861..b93dcc503 100644 --- a/main.py +++ b/main.py @@ -230,6 +230,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 d48bbc282..1a40a8493 100644 --- a/nodes.py +++ b/nodes.py @@ -478,9 +478,12 @@ class SaveLatent: class LoadLatent: @classmethod def INPUT_TYPES(s): - input_dir = folder_paths.get_input_directory() + input_dir = os.path.join(folder_paths.get_output_directory(), "latents") # Adjusts to "Input/Latent" + if not os.path.exists(input_dir): + os.makedirs(input_dir, exist_ok=True) 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" From 23593a6b853a9fb60a805ddfecba6f1e7837753f Mon Sep 17 00:00:00 2001 From: Vijay Date: Sun, 10 Nov 2024 10:13:26 +0530 Subject: [PATCH 2/2] Removed unwanted code --- nodes.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/nodes.py b/nodes.py index 1a40a8493..267ca44ac 100644 --- a/nodes.py +++ b/nodes.py @@ -478,10 +478,6 @@ class SaveLatent: class LoadLatent: @classmethod def INPUT_TYPES(s): - input_dir = os.path.join(folder_paths.get_output_directory(), "latents") # Adjusts to "Input/Latent" - if not os.path.exists(input_dir): - os.makedirs(input_dir, exist_ok=True) - 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": (folder_paths.get_filename_list("latents"), {"tooltip": "The name of the latent from output/latents to load."})}}