Files

22 lines
903 B
Python

from .nodes import Text2VideoNode, ImageGeneratorNode, TextToAudioNode, Video2AudioNode, ImageExpanderNode, \
Image2VideoNode, KLingAIAPIClient, PreviewVideo, PreviewAudio, KolorsVirtualTryOnNode, VideoExtendNode, LipSyncNode, \
LipSyncTextInputNode, LipSyncAudioInputNode, EffectNode
NODE_CLASS_MAPPINGS = {
'Client': KLingAIAPIClient,
'Image Generator': ImageGeneratorNode,
'Image Expander': ImageExpanderNode,
"Text2Audio": TextToAudioNode,
"Video2Audio": Video2AudioNode,
'Text2Video': Text2VideoNode,
'Image2Video': Image2VideoNode,
'Virtual Try On': KolorsVirtualTryOnNode,
'KLingAI Preview Video': PreviewVideo,
'KLingAI Preview Audio': PreviewAudio,
'Video Extender': VideoExtendNode,
'Lip Sync': LipSyncNode,
'Lip Sync Text Input': LipSyncTextInputNode,
'Lip Sync Audio Input': LipSyncAudioInputNode,
'Effects': EffectNode
}