Browse Source

fix loading alt folders

pull/425/head
sALTaccount 2 years ago
parent
commit
7734d65f22
  1. 13
      nodes.py

13
nodes.py

@ -224,10 +224,9 @@ class DiffusersLoader:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
paths = [] paths = []
search_paths = folder_paths.get_folder_paths("diffusers") for search_path in folder_paths.get_folder_paths("diffusers"):
for search_path in search_paths:
if os.path.exists(search_path): if os.path.exists(search_path):
paths = next(os.walk(search_path))[1] 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"
@ -235,7 +234,13 @@ class DiffusersLoader:
CATEGORY = "loaders" CATEGORY = "loaders"
def load_checkpoint(self, model_path, output_vae=True, output_clip=True): def load_checkpoint(self, model_path, output_vae=True, output_clip=True):
model_path = os.path.join(folder_paths.models_dir, 'diffusers', model_path) for search_path in folder_paths.get_folder_paths("diffusers"):
if os.path.exists(search_path):
paths = next(os.walk(search_path))[1]
if model_path in paths:
model_path = os.path.join(search_path, model_path)
break
search_paths = folder_paths.get_folder_paths("diffusers")
return load_diffusers(model_path, fp16=model_management.should_use_fp16(), output_vae=output_vae, output_clip=output_clip, embedding_directory=folder_paths.get_folder_paths("embeddings")) return load_diffusers(model_path, fp16=model_management.should_use_fp16(), output_vae=output_vae, output_clip=output_clip, embedding_directory=folder_paths.get_folder_paths("embeddings"))

Loading…
Cancel
Save