Skip to content

Commit 3a49dab

Browse files
committed
Update: stream_bert_vits2_infer
1 parent 7943e2f commit 3a49dab

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

manager/TTSManager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -425,8 +425,8 @@ def bert_vits2_infer(self, state, encode=True):
425425
return self.encode(sampling_rate, audio, state["format"]) if encode else audio
426426

427427
def stream_bert_vits2_infer(self, state, encode=True):
428-
model = self.get_model(ModelType.BERT_VITS2, state["id"])
429-
state["id"] = self.get_real_id(ModelType.BERT_VITS2, state["id"])
428+
model = self.get_model(model_type=ModelType.BERT_VITS2, id=state["id"], speaker=state["speaker"])
429+
state["id"] = self.get_real_id(model_type=ModelType.BERT_VITS2, id=state["id"], speaker=state["speaker"])
430430

431431
# 去除所有多余的空白字符
432432
if state["text"] is not None:

0 commit comments

Comments
 (0)