Add tooltips to LoraSave node.

Comfy-Org · Oct 18, 2024 · 518c0dc · 518c0dc
1 parent ce0542e
commit 518c0dc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/comfy_extras/nodes_lora_extract.py b/comfy_extras/nodes_lora_extract.py
@@ -82,8 +82,8 @@ def INPUT_TYPES(s):
                               "lora_type": (tuple(LORA_TYPES.keys()),),
                               "bias_diff": ("BOOLEAN", {"default": True}),
                             },
-                "optional": {"model_diff": ("MODEL",),
-                             "text_encoder_diff": ("CLIP",)},
+                "optional": {"model_diff": ("MODEL", {"tooltip": "The ModelSubtract output to be converted to a lora."}),
+                             "text_encoder_diff": ("CLIP", {"tooltip": "The CLIPSubtract output to be converted to a lora."})},
     }
     RETURN_TYPES = ()
     FUNCTION = "save"
@@ -113,3 +113,7 @@ def save(self, filename_prefix, rank, lora_type, bias_diff, model_diff=None, tex
 NODE_CLASS_MAPPINGS = {
     "LoraSave": LoraSave
 }
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "LoraSave": "Extract and Save Lora"
+}