mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 05:32:12 -03:00
2eea92abdf
Use union type "AUTOCOMPLETE_TEXT_PROMPT,STRING" to enable input mode compatibility with STRING outputs while preserving autocomplete widget functionality via widgetType option. Fixes issue where text inputs could not receive connections from STRING-type outputs after changing from built-in STRING to custom AUTOCOMPLETE_TEXT_PROMPT type. Affected nodes: - Prompt (LoraManager) - Text (LoraManager)
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from typing import Any, Optional
|
|
|
|
class PromptLM:
|
|
"""Encodes text (and optional trigger words) into CLIP conditioning."""
|
|
|
|
NAME = "Prompt (LoraManager)"
|
|
CATEGORY = "Lora Manager/conditioning"
|
|
DESCRIPTION = (
|
|
"Encodes a text prompt using a CLIP model into an embedding that can be used "
|
|
"to guide the diffusion model towards generating specific images."
|
|
)
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"text": (
|
|
"AUTOCOMPLETE_TEXT_PROMPT,STRING",
|
|
{
|
|
"widgetType": "AUTOCOMPLETE_TEXT_PROMPT",
|
|
"placeholder": "Enter prompt... /char, /artist for quick tag search",
|
|
"tooltip": "The text to be encoded.",
|
|
},
|
|
),
|
|
"clip": (
|
|
'CLIP',
|
|
{"tooltip": "The CLIP model used for encoding the text."},
|
|
),
|
|
},
|
|
"optional": {
|
|
"trigger_words": (
|
|
'STRING',
|
|
{
|
|
"forceInput": True,
|
|
"tooltip": (
|
|
"Optional trigger words to prepend to the text before "
|
|
"encoding."
|
|
)
|
|
},
|
|
)
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ('CONDITIONING', 'STRING',)
|
|
RETURN_NAMES = ('CONDITIONING', 'PROMPT',)
|
|
OUTPUT_TOOLTIPS = (
|
|
"A conditioning containing the embedded text used to guide the diffusion model.",
|
|
)
|
|
FUNCTION = "encode"
|
|
|
|
def encode(self, text: str, clip: Any, trigger_words: Optional[str] = None):
|
|
prompt = text
|
|
if trigger_words:
|
|
prompt = ", ".join([trigger_words, text])
|
|
|
|
from nodes import CLIPTextEncode # type: ignore
|
|
conditioning = CLIPTextEncode().encode(clip, prompt)[0]
|
|
return (conditioning, prompt,) |