qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

allow configurable path for diffusers models

This commit is contained in:
sALTaccount 2023-04-06 21:45:08 -07:00
parent 42fd67b5cb
commit 72a8973bd5
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
folder_paths.py
View File

@ -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)

2
nodes.py
View File

@ -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,), }}