From a21bafaf743b44455be815bea45cd66458da6d26 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Thu, 25 Jul 2024 03:26:37 +0800
Subject: [PATCH] fix(core): missig param

---
 ChatTTS/core.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ChatTTS/core.py b/ChatTTS/core.py
index 26b780d95..5824e8f96 100644
--- 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)