|
|
|
@ -224,10 +224,9 @@ class DiffusersLoader:
|
|
|
|
|
@classmethod
|
|
|
|
|
def INPUT_TYPES(cls):
|
|
|
|
|
paths = []
|
|
|
|
|
search_paths = folder_paths.get_folder_paths("diffusers")
|
|
|
|
|
for search_path in search_paths:
|
|
|
|
|
for search_path in folder_paths.get_folder_paths("diffusers"):
|
|
|
|
|
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_TYPES = ("MODEL", "CLIP", "VAE")
|
|
|
|
|
FUNCTION = "load_checkpoint"
|
|
|
|
@ -235,7 +234,13 @@ class DiffusersLoader:
|
|
|
|
|
CATEGORY = "loaders"
|
|
|
|
|
|
|
|
|
|
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"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|