diff --git a/CheckpointLoaderSimpleText.py b/CheckpointLoaderSimpleText.py index 6cd3f9b..3bdf13e 100644 --- a/CheckpointLoaderSimpleText.py +++ b/CheckpointLoaderSimpleText.py @@ -37,6 +37,9 @@ class CheckpointLoaderSimpleText: def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True): print(f"[{ccolor}]ckpt_name : [/{ccolor}]", ckpt_name) ckpt_path=folder_paths.get_full_path("checkpoints", ckpt_name) + if ckpt_path is None: + ckpt_path=getFullPath(ckpt_name,"checkpoints") + print(f"[{ccolor}]ckpt_path : [/{ccolor}]", ckpt_path) try: out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings")) return out