From 766ae119a86e51bf0ee0b068c26fa3acc0e1b815 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 19 Aug 2024 14:00:56 -0400
Subject: [PATCH] CheckpointSave node name.

---
 comfy_extras/nodes_model_merging.py | 4 ++++
 1 file changed, 4 insertions(+)

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",
+}