diffusion/__init__.py

from typing import Union

from .diffusion import SpacedDiffusionBeatGans, SpacedDiffusionBeatGansConfig

Sampler = Union[SpacedDiffusionBeatGans]
SamplerConfig = Union[SpacedDiffusionBeatGansConfig]