45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
from .lumaai_api_node import (
|
|
LumaAIClient,
|
|
Text2Video,
|
|
Image2Video,
|
|
InterpolateGenerations,
|
|
ExtendGeneration,
|
|
PreviewVideo,
|
|
Reference,
|
|
ConcatReferences,
|
|
CharacterReference,
|
|
ImageGeneration,
|
|
ModifyImage,
|
|
)
|
|
from .imgbb_node import ImgBBUpload
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LumaAIClient": LumaAIClient,
|
|
"ImgBBUpload": ImgBBUpload,
|
|
"LumaText2Video": Text2Video,
|
|
"LumaImage2Video": Image2Video,
|
|
"LumaInterpolateGenerations": InterpolateGenerations,
|
|
"LumaExtendGeneration": ExtendGeneration,
|
|
"LumaPreviewVideo": PreviewVideo,
|
|
"Reference": Reference,
|
|
"ConcatReferences": ConcatReferences,
|
|
"CharacterReference": CharacterReference,
|
|
"LumaImageGeneration": ImageGeneration,
|
|
"LumaModifyImage": ModifyImage,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LumaAIClient": "LumaAI Client",
|
|
"ImgBBUpload": "ImgBB Upload",
|
|
"LumaText2Video": "Text to Video",
|
|
"LumaImage2Video": "Image to Video",
|
|
"LumaInterpolateGenerations": "Interpolate Generations",
|
|
"LumaExtendGeneration": "Extend Generation",
|
|
"LumaPreviewVideo": "LumaAI Preview Video",
|
|
"LumaReference": "Reference",
|
|
"LumaConcatReferences": "Concat References",
|
|
"LumaCharacterReference": "Character Reference",
|
|
"LumaImageGeneration": "Image Generation",
|
|
"LumaModifyImage": "Modify Image",
|
|
}
|