Merge pull request #133 from zhuzizyf/fix-frontend-bug

Update frontend.py
This commit is contained in:
Xiang Lyu
2024-07-15 10:01:51 +08:00
committed by GitHub

View File

@@ -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)]