models/__init__.py

# ViuAI_TTS_200M Models Package
from .text_encoder import TextEncoder
from .dit_backbone import DiTBackbone
from .duration_predictor import DurationPredictor
from .flow_matching import OptimalTransportCFM
from .vocoder import NeuralVocoder
from .viuai_tts import ViuAITTS200M

__all__ = [
    "TextEncoder",
    "DiTBackbone",
    "DurationPredictor",
    "OptimalTransportCFM",
    "NeuralVocoder",
    "ViuAITTS200M",
]