| 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] | |