from typing import Union

from visualizr.diffusion.diffusion import (
    SpacedDiffusionBeatGans,
    SpacedDiffusionBeatGansConfig,
)

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