Parcourir la source

Merge branch 'main' of github.com:FunAudioLLM/CosyVoice into main

lyuxiang.lx il y a 1 an
Parent
commit
a723ea375e
1 fichiers modifiés avec 4 ajouts et 1 suppressions
  1. 4 1
      cosyvoice/cli/frontend.py

+ 4 - 1
cosyvoice/cli/frontend.py

@@ -114,7 +114,10 @@ class CosyVoiceFrontEnd:
                                                 token_min_n=60, merge_len=20,
                                                 comma_split=False)]
         else:
-            text = self.en_tn_model.normalize(text)
+            if self.use_ttsfrd:
+                text = self.frd.get_frd_extra_info(text, 'input')
+            else:
+                text = self.en_tn_model.normalize(text)
             text = spell_out_number(text, self.inflect_parser)
             texts = [i for i in split_paragraph(text, partial(self.tokenizer.encode, allowed_special=self.allowed_special), "en", token_max_n=80,
                                                 token_min_n=60, merge_len=20,