From 9ee632bb9630c80fe69ef0558c1cb1a719c73995 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Wed, 3 Jan 2024 12:50:43 +0900
Subject: [PATCH] Improve: style init validation check

---
 common/tts_model.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/common/tts_model.py b/common/tts_model.py
index 109c63c..05c32c3 100644
--- a/common/tts_model.py
+++ b/common/tts_model.py
@@ -35,13 +35,23 @@ class Model:
         self.hps: utils.HParams = utils.get_hparams_from_file(self.config_path)
         self.spk2id: Dict[str, int] = self.hps.data.spk2id
         self.id2spk: Dict[int, str] = {v: k for k, v in self.spk2id.items()}
+
         self.num_styles: int = self.hps.data.num_styles
         if hasattr(self.hps.data, "style2id"):
             self.style2id: Dict[str, int] = self.hps.data.style2id
         else:
             self.style2id: Dict[str, int] = {str(i): i for i in range(self.num_styles)}
+        if len(self.style2id) != self.num_styles:
+            raise ValueError(
+                f"Number of styles ({self.num_styles}) does not match the number of style2id ({len(self.style2id)})"
+            )
 
         self.style_vectors: np.ndarray = np.load(self.style_vec_path)
+        if self.style_vectors.shape[0] != self.num_styles:
+            logger.warning(
+                f"The number of styles ({self.num_styles}) does not match the number of style vectors ({self.style_vectors.shape[0]})"
+            )
+
         self.net_g: Optional[SynthesizerTrn] = None
 
     def load_net_g(self):