From 72a8973bd56b7cc179eb603ccd61385fdca5766d Mon Sep 17 00:00:00 2001 From: sALTaccount Date: Thu, 6 Apr 2023 21:45:08 -0700 Subject: [PATCH] allow configurable path for diffusers models --- folder_paths.py | 1 + nodes.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/folder_paths.py b/folder_paths.py index f13e4895..ab335934 100644 --- a/folder_paths.py +++ b/folder_paths.py @@ -23,6 +23,7 @@ folder_names_and_paths["clip"] = ([os.path.join(models_dir, "clip")], supported_ folder_names_and_paths["clip_vision"] = ([os.path.join(models_dir, "clip_vision")], supported_pt_extensions) folder_names_and_paths["style_models"] = ([os.path.join(models_dir, "style_models")], supported_pt_extensions) folder_names_and_paths["embeddings"] = ([os.path.join(models_dir, "embeddings")], supported_pt_extensions) +folder_names_and_paths["diffusers"] = ([os.path.join(models_dir, "diffusers")], ["folder"]) folder_names_and_paths["controlnet"] = ([os.path.join(models_dir, "controlnet"), os.path.join(models_dir, "t2i_adapter")], supported_pt_extensions) folder_names_and_paths["upscale_models"] = ([os.path.join(models_dir, "upscale_models")], supported_pt_extensions) diff --git a/nodes.py b/nodes.py index 8271da04..934b458f 100644 --- a/nodes.py +++ b/nodes.py @@ -224,7 +224,7 @@ class DiffusersLoader: @classmethod def INPUT_TYPES(cls): paths = [] - search_path = os.path.join(folder_paths.models_dir, 'diffusers') + search_path = folder_paths.get_folder_paths("diffusers")[0] if os.path.exists(search_path): paths = next(os.walk(search_path))[1] return {"required": {"model_path": (paths,), }}