DisMouse/diffusion/__init__.py

7 lines
199 B
Python
Raw Permalink Normal View History

2024-10-08 14:18:47 +02:00
from typing import Union
from .diffusion import SpacedDiffusionBeatGans, SpacedDiffusionBeatGansConfig
Sampler = Union[SpacedDiffusionBeatGans]
SamplerConfig = Union[SpacedDiffusionBeatGansConfig]