CheckpointSave node name.

6 months ago · 766ae119a8
parent fc90ceb6ba
commit 766ae119a8
1 changed files with 4 additions and 0 deletions
--- 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",
+}