wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-12 17:27:26 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Merge 9695bc691e into f48d2a017e

Browse Source
This commit is contained in:
Varun Raj 2026-05-18 12:02:38 +08:00 committed by GitHub
commit 1dbf84c889
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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"] SEARCH_ALIASES = ["load model", "checkpoint", "model loader", "load checkpoint", "ckpt", "model"]
def load_checkpoint(self, ckpt_name): def load_checkpoint(self, ckpt_name):
ckpt_path = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name) try:
out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings")) 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] return out[:3]
class DiffusersLoader: class DiffusersLoader: