from medvae.models.autoencoder_kl_3d import AutoencoderKL as AutoencoderKL_3D
from medvae.models.autoencoder_kl import AutoencoderKL as AutoencoderKL_2D

__all__ = ["AutoencoderKL_2D", "AutoencoderKL_3D"]