fix frontend bug, test=tts

3 years ago · 34eb41a2a5
parent 8ea289a251
commit 34eb41a2a5
1 changed files with 2 additions and 1 deletions
--- a/paddlespeech/t2s/frontend/g2pw/onnx_api.py
+++ b/paddlespeech/t2s/frontend/g2pw/onnx_api.py
@ -210,7 +210,8 @@ class G2PWOnnxConverter:
        for sent_id, sent in enumerate(sentences):
            # pypinyin works well for Simplified Chinese than Traditional Chinese
            sent_s = tranditional_to_simplified(sent)
-            pypinyin_result = pinyin(sent_s, style=Style.TONE3)
+            pypinyin_result = pinyin(
+                sent_s, neutral_tone_with_five=True, style=Style.TONE3)
            partial_result = [None] * len(sent)
            for i, char in enumerate(sent):
                if char in self.polyphonic_chars_new: