Fix: stream_bert_vits2_infer lang error

Artrajz · Nov 27, 2024 · a13f030 · a13f030
1 parent 4838262
commit a13f030
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/manager/TTSManager.py b/manager/TTSManager.py
@@ -426,10 +426,10 @@ def stream_bert_vits2_infer(self, state, encode=True):
         sampling_rate = model.sampling_rate
         sentences_list = sentence_split(state["text"], state["segment_size"])
 
-        if state["lang"].lower() == "auto":
-            infer_func = model.infer_multilang
-        else:
+        if len(state["lang"]) == 1 and "auto" not in state["lang"]:
             infer_func = model.infer
+        else:
+            infer_func = model.infer_multilang
 
         for sentences in sentences_list:
             state["text"] = sentences