capspeech/nar/model/__init__.py

from capspeech.nar.model.cfm import CFM

from capspeech.nar.model.backbones.unett import UNetT
from capspeech.nar.model.backbones.dit import DiT

from capspeech.nar.model.trainer import Trainer