from typing import Union
from .DiffAE_model_unet import BeatGANsUNetModel, BeatGANsUNetConfig
from .DiffAE_model_unet_autoenc import BeatGANsAutoencConfig, BeatGANsAutoencModel
from .DiffAE_model_latentnet import *

Model = Union[BeatGANsUNetModel, BeatGANsAutoencModel]
ModelConfig = Union[BeatGANsUNetConfig, BeatGANsAutoencConfig]