WaveGen / src /wavegen_v_spec /model /__init__.py
LeeSangHoon's picture
Add ZeroGPU demo with preloaded V-Spec and Flash
8d6ad36 verified
Raw History Blame Contribute Delete
164 Bytes
"""WaveGen generator with the V-Spec training objective."""
from .cfm import CFM, CFMMSVSpec
from .backbones.dit import DiT
__all__ = ["CFM", "CFMMSVSpec", "DiT"]