f5_tts/model_new/__init__.py

from f5_tts.model_new.backbones.dit import DiT
from f5_tts.model_new.backbones.mmdit import MMDiT
from f5_tts.model_new.backbones.unett import UNetT
from f5_tts.model_new.cfm import CFM
from f5_tts.model_new.trainer import Trainer

__all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]