Fix the passing of the lang parameter

Artrajz · Jul 19, 2024 · 253faa7 · 253faa7
1 parent 2e44d31
commit 253faa7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/gpt_sovits/gpt_sovits.py b/gpt_sovits/gpt_sovits.py
@@ -263,7 +263,13 @@ def preprocess_text(self, text: str, lang: str, segment_size: int):
                 text = "。" + text if lang != "en" else "." + text
             if (text[-1] not in splits):
                 text += "。" if lang != "en" else "."
-            phones, word2ph, norm_text, bert_features = self.get_bert_and_cleaned_text_multilang(text)
+
+            if lang == "auto":
+                phones, word2ph, norm_text, bert_features = self.get_bert_and_cleaned_text_multilang(text)
+            else:
+                phones, word2ph, norm_text = self.get_cleaned_text(text, lang)
+                bert_features = self.get_bert_feature(text, phones, word2ph, norm_text).to(self.device)
+
             res = {
                 "phones": phones,
                 "bert_features": bert_features,