29 lines
855 B
Python
29 lines
855 B
Python
"""ComfyUI-MVNT: AI dance choreography from music."""
|
|
|
|
from .nodes import (
|
|
MVNTAudioSegment,
|
|
MVNTImageToTPose,
|
|
MVNTGenerateDance,
|
|
MVNTRenderDanceVideo,
|
|
MVNTPreviewDance3D,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MVNT Audio Segment": MVNTAudioSegment,
|
|
"MVNT Image to T-Pose": MVNTImageToTPose,
|
|
"MVNT Generate Dance": MVNTGenerateDance,
|
|
"MVNT Render Dance Video": MVNTRenderDanceVideo,
|
|
"MVNTPreviewDance3D": MVNTPreviewDance3D,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MVNT Audio Segment": "MVNT Audio Segment",
|
|
"MVNT Image to T-Pose": "MVNT Image to T-Pose",
|
|
"MVNT Generate Dance": "MVNT Generate Dance",
|
|
"MVNT Render Dance Video": "MVNT Render Dance Video",
|
|
"MVNTPreviewDance3D": "MVNT Preview Dance 3D",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|