刘悦 a3b108bfe6
Add files via upload
添加resemble-enhance源文件
2024-03-11 14:18:49 +08:00

24 lines
557 B
Python

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