add a concat text build-in node

This commit is contained in:
AhBumm
2024-11-05 13:27:16 +08:00
parent cebca2204b
commit c4ec7a2820
2 changed files with 29 additions and 1 deletions
+5
View File
@@ -13,6 +13,11 @@ NODE_CLASS_MAPPINGS = {
"BillBum_Modified_Structured_LLM_Node(Imperfect)": BillBum_Modified_Structured_LLM_Node,
"BillBum_Modified_Flux_API_Node": BillBum_Modified_Flux_API_Node,
"BillBum_Modified_RegText_Node": BillBum_Modified_RegText_Node,
"BillBum_Modified_DropoutToken_Node": BillBum_Modified_DropoutToken_Node,
"BillBum_Modified_Image_API_Call_Node": BillBum_Modified_Image_API_Call_Node,
"BillBum_Modified_Recraft_API_Node": BillBum_Modified_Recraft_API_Node,
"Text_Concat": Text_Concat,
"Input_Text": Input_Text,
}
NODE_DISPLAY_NAME_MAPPINGS = {
+24 -1
View File
@@ -76,6 +76,29 @@ def pil2tensor(images: Image.Image | list[Image.Image]) -> torch.Tensor:
else:
return torch.cat([single_pil2tensor(img) for img in images], dim=0)
class Text_Concat:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"text_1": ("STRING", {"defaultInput": True},),
"text_2": ("STRING", {"defaultInput": True},),
},
}
RETURN_TYPES = ("STRING",)
RETURN_NAMES = ("text",)
FUNCTION = "text_concat"
CATEGORY = "string processing"
def text_concat(self, text_1, text_2):
text = [text_1, " ", text_2]
return (text,)
class Input_Text:
def __init__(self):
@@ -85,7 +108,7 @@ class Input_Text:
def INPUT_TYPES(s):
return {
"required": {
"text": ("STRING",),
"text": ("STRING", {"multiline": True},),
},
}