298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

search all diffusers paths (oops)

Browse Source
This commit is contained in:
sALTaccount 2023-04-06 21:48:58 -07:00
parent 72a8973bd5
commit f51b7a92c7
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@ -224,9 +224,10 @@ class DiffusersLoader:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
paths = [] paths = []
search_path = folder_paths.get_folder_paths("diffusers")[0] search_paths = folder_paths.get_folder_paths("diffusers")
if os.path.exists(search_path): for search_path in search_paths:
paths = next(os.walk(search_path))[1] if os.path.exists(search_path):
paths = next(os.walk(search_path))[1]
return {"required": {"model_path": (paths,), }} return {"required": {"model_path": (paths,), }}
RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_TYPES = ("MODEL", "CLIP", "VAE")
FUNCTION = "load_checkpoint" FUNCTION = "load_checkpoint"