Latent files saved under Input/Latent but require manual move to Inpu…

…t for loading

main.py

@@ -230,6 +230,8 @@ def cleanup_temp():
     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)

nodes.py

@@ -478,9 +478,12 @@ def save(self, samples, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=No
 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"