Update CheckpointLoaderSimpleText.py

2023-08-04 23:58:51 +09:00
parent b4191ce40f
commit c34ebf1940
1 changed files with 3 additions and 0 deletions
@@ -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