Browse Source

empty list if diffusers directory doesn't exist

pull/425/head
sALTaccount 2 years ago
parent
commit
3d16077e38
  1. 7
      nodes.py

7
nodes.py

@ -223,8 +223,11 @@ class CheckpointLoaderSimple:
class DiffusersLoader: class DiffusersLoader:
@classmethod @classmethod
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return {"required": {"model_path": (os.listdir(os.path.join(folder_paths.models_dir, 'diffusers'), ),), paths = []
}} search_path = os.path.join(folder_paths.models_dir, 'diffusers')
if os.path.exists(search_path):
paths = next(os.walk(search_path))[1]
return {"required": {"model_path": (paths,), }}
RETURN_TYPES = ("MODEL", "CLIP", "VAE") RETURN_TYPES = ("MODEL", "CLIP", "VAE")
FUNCTION = "load_checkpoint" FUNCTION = "load_checkpoint"

Loading…
Cancel
Save