update readme

This commit is contained in:
lyuxiang.lx
2024-12-16 14:05:00 +08:00
parent d43a0171d4
commit 1298d90e48
4 changed files with 19 additions and 4 deletions

View File

@@ -109,6 +109,10 @@ class CosyVoiceFrontEnd:
def text_normalize(self, text, split=True):
text = text.strip()
# NOTE(lyuxiang.lx) move this judgement into ttsfrd in the future
for token in self.tokenizer.special_tokens['additional_special_tokens']:
if token in text:
return text if split is False else [text]
if contains_chinese(text):
if self.use_ttsfrd:
texts = [i["text"] for i in json.loads(self.frd.do_voicegen_frd(text))["sentences"]]