Browse Source

allow configurable path for diffusers models

pull/425/head
sALTaccount 2 years ago
parent
commit
72a8973bd5
  1. 1
      folder_paths.py
  2. 2
      nodes.py

1
folder_paths.py

@ -23,6 +23,7 @@ folder_names_and_paths["clip"] = ([os.path.join(models_dir, "clip")], supported_
folder_names_and_paths["clip_vision"] = ([os.path.join(models_dir, "clip_vision")], supported_pt_extensions)
folder_names_and_paths["style_models"] = ([os.path.join(models_dir, "style_models")], supported_pt_extensions)
folder_names_and_paths["embeddings"] = ([os.path.join(models_dir, "embeddings")], supported_pt_extensions)
folder_names_and_paths["diffusers"] = ([os.path.join(models_dir, "diffusers")], ["folder"])
folder_names_and_paths["controlnet"] = ([os.path.join(models_dir, "controlnet"), os.path.join(models_dir, "t2i_adapter")], supported_pt_extensions)
folder_names_and_paths["upscale_models"] = ([os.path.join(models_dir, "upscale_models")], supported_pt_extensions)

2
nodes.py

@ -224,7 +224,7 @@ class DiffusersLoader:
@classmethod
def INPUT_TYPES(cls):
paths = []
search_path = os.path.join(folder_paths.models_dir, 'diffusers')
search_path = folder_paths.get_folder_paths("diffusers")[0]
if os.path.exists(search_path):
paths = next(os.walk(search_path))[1]
return {"required": {"model_path": (paths,), }}

Loading…
Cancel
Save