Fix: add use_jp_extra parameter to config.json (for non-jp-extra trai…

…ning)
litagin02 · Mar 11, 2024 · b3cc705 · b3cc705
1 parent 44851a6
commit b3cc705
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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,