diff --git a/configs/config.json b/configs/config.json index 2f2ce7f..b8082bc 100644 --- a/configs/config.json +++ b/configs/config.json @@ -24,6 +24,7 @@ "freeze_encoder": false }, "data": { + "use_jp_extra": false, "training_files": "Data/your_model_name/filelists/train.list", "validation_files": "Data/your_model_name/filelists/val.list", "max_wav_value": 32768.0, diff --git a/style_bert_vits2/models/hyper_parameters.py b/style_bert_vits2/models/hyper_parameters.py index 30c579e..254b285 100644 --- a/style_bert_vits2/models/hyper_parameters.py +++ b/style_bert_vits2/models/hyper_parameters.py @@ -16,7 +16,7 @@ class HyperParametersTrain(BaseModel): seed: int = 42 epochs: int = 1000 learning_rate: float = 0.0001 - betas: list[float] = [0.8, 0.99] + betas: tuple[float, float] = (0.8, 0.99) eps: float = 1e-9 batch_size: int = 2 bf16_run: bool = False @@ -50,7 +50,7 @@ class HyperParametersData(BaseModel): mel_fmin: float = 0.0 mel_fmax: Optional[float] = None add_blank: bool = True - n_speakers: int = 512 + n_speakers: int = 1 cleaned_text: bool = True spk2id: dict[str, int] = { "Dummy": 0,