From b3cc705caa1d8572cc41d6906d6511c4f1e41709 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Mon, 11 Mar 2024 17:51:55 +0900
Subject: [PATCH] Fix: add use_jp_extra parameter to config.json (for
 non-jp-extra training)

---
 configs/config.json                         | 1 +
 style_bert_vits2/models/hyper_parameters.py | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

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,