Fix GPT-SOVITS in ssml

Artrajz · Jun 6, 2024 · 1afc101 · 1afc101
1 parent 3c5d6b4
commit 1afc101
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/manager/ModelManager.py b/manager/ModelManager.py
@@ -280,6 +280,8 @@ def _load_model_from_path(self, model_path, config_path, sovits_path, gpt_path):
             self.model_handler.load_bert("CHINESE_ROBERTA_WWM_EXT_LARGE")
             model.load_model(self.model_handler)
 
+            self.available_tts_model.add(ModelType.GPT_SOVITS.value)
+
         sid2model = []
         speakers = []
         new_id = len(self.voice_speakers[model_type.value])