mogen/models/architectures/__init__.py

from .vae_architecture import MotionVAE
from .diffusion_architecture import MotionDiffusion

__all__ = [
    'MotionVAE', 'MotionDiffusion'
]