diff --git a/comfy_extras/nodes_textgen.py b/comfy_extras/nodes_textgen.py index 82a0fdb9d..d52faf815 100644 --- a/comfy_extras/nodes_textgen.py +++ b/comfy_extras/nodes_textgen.py @@ -26,6 +26,7 @@ class TextGenerate(io.ComfyNode): return io.Schema( node_id="TextGenerate", + display_name="Generate Text", category="text", search_aliases=["LLM", "gemma"], inputs=[ @@ -157,6 +158,7 @@ class TextGenerateLTX2Prompt(TextGenerate): parent_schema = super().define_schema() return io.Schema( node_id="TextGenerateLTX2Prompt", + display_name="Generate LTX2 Prompt", category=parent_schema.category, inputs=parent_schema.inputs, outputs=parent_schema.outputs,