This commit is contained in:
tuna2134
2026-07-20 21:08:44 +09:00
parent 66de777e06
commit 0c2be00f0a
16 changed files with 439 additions and 7 deletions

View File

@@ -27,6 +27,7 @@ class HyperParametersTrain(BaseModel):
warmup_epochs: int = 0
c_mel: int = 45
c_kl: float = 1.0
c_matcha: float = 1.0
c_commit: int = 100
skip_optimizer: bool = False
freeze_ZH_bert: bool = False
@@ -76,6 +77,12 @@ class HyperParametersModel(BaseModel):
use_mel_posterior_encoder: bool = False
use_duration_discriminator: bool = False
use_wavlm_discriminator: bool = True
use_matcha: bool = False
matcha_channels: int = 192
matcha_num_heads: int = 2
matcha_dropout: float = 0.05
matcha_sigma_min: float = 0.0001
matcha_n_timesteps: int = 8
inter_channels: int = 192
hidden_channels: int = 192
filter_channels: int = 768