Skip to content

GitLab

Commit f51b7a92 authored by sALTaccount's avatar sALTaccount
Browse files

search all diffusers paths (oops)

parent 72a8973b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
nodes.py
View file @ f51b7a92
...@@ -224,9 +224,10 @@ class DiffusersLoader: ...@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment