diff --git a/comfy_extras/nodes_lora_extract.py b/comfy_extras/nodes_lora_extract.py index 3c2f179d..dfd4fe9f 100644 --- a/comfy_extras/nodes_lora_extract.py +++ b/comfy_extras/nodes_lora_extract.py @@ -82,8 +82,8 @@ class LoraSave: "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 @@ class LoraSave: NODE_CLASS_MAPPINGS = { "LoraSave": LoraSave } + +NODE_DISPLAY_NAME_MAPPINGS = { + "LoraSave": "Extract and Save Lora" +}