Files
ComfyUI-IndexTTS2/__init__.py
2025-10-13 20:15:52 +13:00

27 lines
976 B
Python

from .nodes.indextts2_node import IndexTTS2Simple
from .nodes.indextts2_node_advanced import IndexTTS2Advanced
from .nodes.indextts2_node_emovec import IndexTTS2EmotionVector
from .nodes.indextts2_node_emotext import IndexTTS2EmotionFromText
from .nodes.indextts2_save_audio import IndexTTS2SaveAudio
NODE_CLASS_MAPPINGS = {
"IndexTTS2Advanced": IndexTTS2Advanced,
"IndexTTS2EmotionFromText": IndexTTS2EmotionFromText,
"IndexTTS2EmotionVector": IndexTTS2EmotionVector,
"IndexTTS2SaveAudio": IndexTTS2SaveAudio,
"IndexTTS2Simple": IndexTTS2Simple,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"IndexTTS2Advanced": "IndexTTS2 Advanced",
"IndexTTS2EmotionFromText": "IndexTTS2 Emotion From Text",
"IndexTTS2EmotionVector": "IndexTTS2 Emotion Vector",
"IndexTTS2SaveAudio": "IndexTTS2 Save Audio",
"IndexTTS2Simple": "IndexTTS2 Simple",
}
WEB_DIRECTORY = "./web"
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]