🐛 fix infer func

lenML · Jun 20, 2024 · b0de527 · b0de527
1 parent de66e6b
commit b0de527
Showing 1 changed file with 8 additions and 14 deletions.
diff --git a/modules/ChatTTS/ChatTTS/core.py b/modules/ChatTTS/ChatTTS/core.py
@@ -294,22 +294,16 @@ def infer(
         params_refine_text={},
         params_infer_code={"prompt": "[speed_5]"},
         use_decoder=True,
-        do_text_normalization=True,
-        lang=None,
         stream=False,
-        do_homophone_replacement=True,
     ):
         res_gen = self._infer(
-            text,
-            skip_refine_text,
-            refine_text_only,
-            params_refine_text,
-            params_infer_code,
-            use_decoder,
-            do_text_normalization,
-            lang,
-            stream,
-            do_homophone_replacement,
+            text=text,
+            skip_refine_text=skip_refine_text,
+            refine_text_only=refine_text_only,
+            params_refine_text=params_refine_text,
+            params_infer_code=params_infer_code,
+            use_decoder=use_decoder,
+            stream=stream,
         )
         if stream:
             return res_gen
@@ -318,7 +312,7 @@ def infer(
 
     def refiner_prompt(self, text, params_refine_text={}, stream=False):
         return self.infer(
-            text,
+            text=text,
             skip_refine_text=False,
             refine_text_only=True,
             params_refine_text=params_refine_text,