diff --git a/comfy_extras/nodes_model_merging.py b/comfy_extras/nodes_model_merging.py index 8fb8bf79..ccf60115 100644 --- a/comfy_extras/nodes_model_merging.py +++ b/comfy_extras/nodes_model_merging.py @@ -365,3 +365,7 @@ NODE_CLASS_MAPPINGS = { "VAESave": VAESave, "ModelSave": ModelSave, } + +NODE_DISPLAY_NAME_MAPPINGS = { + "CheckpointSave": "Save Checkpoint", +}