Spaces:
Running
Running
File size: 557 Bytes
32b2aaa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from dataclasses import dataclass
from pathlib import Path
from ..hparams import HParams as HParamsBase
@dataclass(frozen=True)
class HParams(HParamsBase):
cfm_solver_method: str = "midpoint"
cfm_solver_nfe: int = 64
cfm_time_mapping_divisor: int = 4
univnet_nc: int = 96
lcfm_latent_dim: int = 64
lcfm_training_mode: str = "ae"
lcfm_z_scale: float = 5
vocoder_extra_dim: int = 32
gan_training_start_step: int | None = 5_000
enhancer_stage1_run_dir: Path | None = None
denoiser_run_dir: Path | None = None
|