From cadadc89cc711618c1555f48bded89338578f92a Mon Sep 17 00:00:00 2001
From: Vijay <idealvijay@live.com>
Date: Sat, 9 Nov 2024 18:38:00 +0530
Subject: [PATCH] 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"