feat: add new model sixian-hailu-mix-se

app.py

@@ -20,12 +20,14 @@ def load_model(model_id):
     model_ckpt_path = os.path.join(model_dir, "model.pth")
     speaker_file_path = os.path.join(model_dir, "speakers.pth")
     language_file_path = os.path.join(model_dir, "language_ids.json")
+    speaker_embedding_file_path = os.path.join(model_dir, "speaker_embs.pth")
 
     temp_config_path = "temp_config.json"
     with open(config_file_path, "r") as f:
         content = f.read()
         content = content.replace("speakers.pth", speaker_file_path)
         content = content.replace("language_ids.json", language_file_path)
+        content = content.replace("speaker_embs.pth", speaker_embedding_file_path)
         f.close()
     with open(temp_config_path, "w") as f:
         f.write(content)

configs/models.yaml

@@ -1,5 +1,16 @@
 sixian-hailu-mix:
   model: ${load_model:formospeech/taiwanese-hakka-tts-sixian-hailu-mix}
+  avalible_dialect:
+    - sixian
+    - hailu
+  speaker_mapping: # display_name: id
+    sixian/female: XF
+    sixian/male: XM
+    hailu/female: HF
+    hailu/male: HM
+
+sixian-hailu-mix-se:
+  model: ${load_model:formospeech/taiwanese-hakka-tts-sixian-hailu-mix-se}
   avalible_dialect:
     - sixian
     - hailu