File size: 334 Bytes
485c2ee
 
 
 
 
 
 
1
2
3
4
5
6
7
8
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]