Update: stream_bert_vits2_infer

Artrajz · Jan 11, 2025 · 3a49dab · 3a49dab
1 parent 7943e2f
commit 3a49dab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/manager/TTSManager.py b/manager/TTSManager.py
@@ -425,8 +425,8 @@ def bert_vits2_infer(self, state, encode=True):
         return self.encode(sampling_rate, audio, state["format"]) if encode else audio
 
     def stream_bert_vits2_infer(self, state, encode=True):
-        model = self.get_model(ModelType.BERT_VITS2, state["id"])
-        state["id"] = self.get_real_id(ModelType.BERT_VITS2, state["id"])
+        model = self.get_model(model_type=ModelType.BERT_VITS2, id=state["id"], speaker=state["speaker"])
+        state["id"] = self.get_real_id(model_type=ModelType.BERT_VITS2, id=state["id"], speaker=state["speaker"])
 
         # 去除所有多余的空白字符
         if state["text"] is not None: