File size: 334 Bytes
c2ced9d |
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]
|