soumickmj's picture
Upload DiffAE
485c2ee verified
raw
history blame contribute delete
334 Bytes
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]