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

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2noise · Jul 27, 2024 · 06b823b · 06b823b
1 parent 2bfb097
commit 06b823b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -635,7 +635,9 @@ def _refine_text(
             input_ids_list = [i.tolist() for i in input_ids]
             del input_ids
 
-            result = gpt.llm.generate(None, sample_params, input_ids_list, params.show_tqdm)
+            result = gpt.llm.generate(
+                None, sample_params, input_ids_list, params.show_tqdm
+            )
             token_ids = []
             hidden_states = []
             for i in result:

diff --git a/ChatTTS/model/velocity/model_runner.py b/ChatTTS/model/velocity/model_runner.py
@@ -571,7 +571,7 @@ def execute_model(
             logprob_i = logprob[i].cpu().tolist()
             tmp_hidden_states = hidden_states[i].cpu()
             if input_tokens[i].shape[-2] != 1:
-                tmp_hidden_states = tmp_hidden_states[-1:,:]
+                tmp_hidden_states = tmp_hidden_states[-1:, :]
             result = SequenceGroupOutput(
                 samples=[
                     SequenceOutput(