models/__init__.py

from .transformer import MotionTransformer
from .gaussian_diffusion import GaussianDiffusion

__all__ = ['MotionTransformer', 'GaussianDiffusion']