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

fix loading alt folders

Browse Source
This commit is contained in:
sALTaccount 2023-04-06 22:02:26 -07:00
parent f51b7a92c7
commit 7734d65f22
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
nodes.py
View File

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