chore(format): run black on dev (#556)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2noise · Jul 9, 2024 · 00b2dfd · 00b2dfd
1 parent 6e18575
commit 00b2dfd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -401,7 +401,7 @@ def _infer(
                 for i in range(wavs.shape[0]):
                     if b[i] > len(wavs[i]):
                         b[i] = len(wavs[i])
-                    new_wavs[i, :b[i]-a[i]] = wavs[i, a[i]:b[i]]
+                    new_wavs[i, : b[i] - a[i]] = wavs[i, a[i] : b[i]]
                 length = b
                 yield new_wavs
             else:
@@ -410,8 +410,8 @@ def _infer(
             for i in range(wavs.shape[0]):
                 a = length[i]
                 b = len(wavs[i])
-                wavs[i, :b-a] = wavs[i, a:]
-                wavs[i, b-a:] = 0
+                wavs[i, : b - a] = wavs[i, a:]
+                wavs[i, b - a :] = 0
             yield wavs
 
     @torch.inference_mode()

diff --git a/examples/cmd/run.py b/examples/cmd/run.py
@@ -45,7 +45,8 @@ def main(texts: List[str], spk: Optional[str] = None, stream=False):
 
     logger.info("Start inference.")
     wavs = chat.infer(
-        texts, stream,
+        texts,
+        stream,
         params_infer_code=ChatTTS.Chat.InferCodeParams(
             spk_emb=spk,
         ),
@@ -57,7 +58,7 @@ def main(texts: List[str], spk: Optional[str] = None, stream=False):
     for index, wav in enumerate(wavs):
         if stream:
             for i, w in enumerate(wav):
-                save_mp3_file(w, (i+1)*1000+index)
+                save_mp3_file(w, (i + 1) * 1000 + index)
             wavs_list.append(wav)
         else:
             save_mp3_file(wav, index)
@@ -82,10 +83,13 @@ def main(texts: List[str], spk: Optional[str] = None, stream=False):
     parser.add_argument(
         "--stream",
         help="Use stream mode",
-        action='store_true',
+        action="store_true",
     )
     parser.add_argument(
-        "texts", help="Original text", default=["YOUR TEXT HERE"], nargs=argparse.REMAINDER,
+        "texts",
+        help="Original text",
+        default=["YOUR TEXT HERE"],
+        nargs=argparse.REMAINDER,
     )
     args = parser.parse_args()
     main(args.texts, args.spk, args.stream)