path: root/models/__init__.py

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

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