wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-15 20:39:48 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Improve error message when checkpoint file is missing

Browse Source
This commit is contained in:
RAJ-VARUN13 2026-05-15 11:51:09 +05:30
parent 77e2ed5e01
commit 9695bc691e
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
nodes.py
View File

@ -604,8 +604,26 @@ class CheckpointLoaderSimple:
SEARCH_ALIASES = ["load model", "checkpoint", "model loader", "load checkpoint", "ckpt", "model"]
def load_checkpoint(self, ckpt_name):
ckpt_path = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name)
out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings"))
try:
ckpt_path = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name)
except FileNotFoundError:
available = folder_paths.get_filename_list("checkpoints")
available_models = "\n".join(f"- {name}" for name in available[:10])
raise FileNotFoundError(
f"Checkpoint '{ckpt_name}' was not found.\n\n"
f"Expected location:\n"
f"ComfyUI/models/checkpoints/\n\n"
f"Available checkpoints:\n"
f"{available_models if available_models else '(none found)'}"
)
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[:3]
class DiffusersLoader: