소스 검색

Update frontend.py

Fix the bug in handling anomalies for synthetic text ending with Chinese and English commas.
zhuyunfeng 1 년 전
부모
커밋
6ae6ba3f77
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      cosyvoice/cli/frontend.py

+ 2 - 0
cosyvoice/cli/frontend.py

@@ -20,6 +20,7 @@ from typing import Callable
 import torchaudio.compliance.kaldi as kaldi
 import torchaudio
 import os
+import re
 import inflect
 try:
     import ttsfrd
@@ -110,6 +111,7 @@ class CosyVoiceFrontEnd:
             text = text.replace(".", "、")
             text = text.replace(" - ", ",")
             text = remove_bracket(text)
+            text = re.sub(r'[,,]+$', '。', text)
             texts = [i for i in split_paragraph(text, partial(self.tokenizer.encode, allowed_special=self.allowed_special), "zh", token_max_n=80,
                                                 token_min_n=60, merge_len=20,
                                                 comma_split=False)]