From 68d4b8d68bce3475711766e8274d30839c7adc9b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Mon, 21 Aug 2023 08:28:34 +0800
Subject: [PATCH] Update train_ms.py

---
 train_ms.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/train_ms.py b/train_ms.py
index 64365f3..828b26d 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -92,14 +92,19 @@ def run(rank, n_gpus, hps):
     else:
         print("Using normal MAS for VITS1")
         use_noise_scaled_mas = False
+    if "use_spk_conditioned_encoder" in hps.model.keys() and hps.model.use_spk_conditioned_encoder == True:
+        if hps.data.n_speakers == 0:
+            raise ValueError("n_speakers must be > 0 when using spk conditioned encoder to train multi-speaker model")
+        use_spk_conditioned_encoder = True
+    else:
+        print("Using normal encoder for VITS1")
+        use_spk_conditioned_encoder = False
 
     net_g = SynthesizerTrn(
         len(symbols),
         hps.data.filter_length // 2 + 1,
         hps.train.segment_size // hps.data.hop_length,
         n_speakers=hps.data.n_speakers,
-        use_spk_conditioned_encoder=use_spk_conditioned_encoder,
-        use_noise_scaled_mas=use_noise_scaled_mas,
         mas_noise_scale_initial = mas_noise_scale_initial,
         noise_scale_delta = noise_scale_delta,
         **hps.model).cuda(rank)