| from typing import Union | |
| from .unet import BeatGANsUNetModel, BeatGANsUNetConfig | |
| from .unet_autoenc import BeatGANsAutoencConfig, BeatGANsAutoencModel | |
| Model = Union[BeatGANsUNetModel, BeatGANsAutoencModel] | |
| ModelConfig = Union[BeatGANsUNetConfig, BeatGANsAutoencConfig] | |