fix(core): missig param

2noise · Jul 24, 2024 · a21bafa · a21bafa
1 parent 6a4c97b
commit a21bafa
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -548,7 +548,9 @@ def _infer_code(
             del_all(logits_processors)
 
             return [
-                GPT.GenerationOutputs(ids=token_ids, hiddens=hidden_states),
+                GPT.GenerationOutputs(
+                    ids=token_ids, hiddens=hidden_states, attentions=[],
+                ),
             ]
 
         emb = gpt(input_ids, text_mask)
@@ -638,7 +640,8 @@ def _refine_text(
             del_all(logits_warpers)
             del_all(logits_processors)
 
-            return GPT.GenerationOutputs(ids=token_ids, hiddens=hidden_states)
+            return GPT.GenerationOutputs(ids=token_ids, hiddens=hidden_states, attentions=[],
+                                        )
 
         emb = gpt(input_ids, text_mask)