7 lines
270 B
Python
7 lines
270 B
Python
|
from typing import Union
|
||
|
|
from .unet import BeatGANsUNetModel, BeatGANsUNetConfig
|
||
|
|
from .unet_autoenc import BeatGANsAutoencConfig, BeatGANsAutoencModel
|
||
|
|
|
||
|
|
Model = Union[BeatGANsUNetModel, BeatGANsAutoencModel]
|
||
|
|
ModelConfig = Union[BeatGANsUNetConfig, BeatGANsAutoencConfig]
|