"""
Model definitions for MNIST Diffusion.
"""

from .unet import UNetMNIST
from .autoencoder import AEModule, Autoencoder