mogen/models/architectures/__init__.py

from .diffusion_architecture import MotionDiffusion, UnifiedMotionDiffusion
from .vae_architecture import MotionVAE


__all__ = ['MotionVAE', 'MotionDiffusion', 'UnifiedMotionDiffusion']