update example & minor fix

autogyro · May 30, 2024 · d2c7d95 · d2c7d95
1 parent 6c52dfa
commit d2c7d95
Show file tree

Hide file tree

Showing 3 changed files with 235 additions and 78 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -16,8 +16,6 @@
 
 logging.basicConfig(level = logging.INFO)
 
-torch._dynamo.config.suppress_errors = True
-torch.set_float32_matmul_precision('medium')
 
 class Chat:
     def __init__(self, ):
@@ -138,6 +136,9 @@ def infer(
 
         assert self.check_model(use_decoder=use_decoder)
 
+        if not isinstance(text, list): 
+            text = [text]
+
         if do_text_normalization:
             for i, t in enumerate(text):
                 _lang = detect_language(t) if lang is None else lang

diff --git a/ChatTTS/utils/infer_utils.py b/ChatTTS/utils/infer_utils.py
@@ -48,7 +48,7 @@ def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor) -> to
 def count_invalid_characters(s):
 
     s = re.sub(r'\[uv_break\]|\[laugh\]|\[lbreak\]', '', s)
-    pattern = re.compile(r'[^\u4e00-\u9fffA-Za-z，。,\. ]')
+    pattern = re.compile(r'[^\u4e00-\u9fffA-Za-z，。、,\. ]')
     non_alphabetic_chinese_chars = pattern.findall(s)
     return set(non_alphabetic_chinese_chars)
 

diff --git a/infer.ipynb → example.ipynb b/infer.ipynb → example.ipynb