From 57bb55baca658b7956c598480b807ab1a93a7702 Mon Sep 17 00:00:00 2001 From: Symbiomatrix Date: Fri, 11 Apr 2025 16:19:23 +0300 Subject: [PATCH] Add input storage. --- folder_paths.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/folder_paths.py b/folder_paths.py index 72c70f594..bf931dd09 100644 --- a/folder_paths.py +++ b/folder_paths.py @@ -10,6 +10,7 @@ from collections.abc import Collection from comfy.cli_args import args supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft'} +supported_input_extensions: set[str] = {""} # {".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".webp", ".webm", ".mp4", ".mkv"} folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {} @@ -51,6 +52,9 @@ temp_directory = os.path.join(base_path, "temp") input_directory = os.path.join(base_path, "input") user_directory = os.path.join(base_path, "user") +folder_names_and_paths["input"] = ([input_directory], supported_input_extensions) +folder_names_and_paths["latent"] = ([input_directory], {".latent"}) + filename_list_cache: dict[str, tuple[list[str], dict[str, float], float]] = {} class CacheHelper: