From decec67b78020abd5d7c5bf22788ee02c04aad2d Mon Sep 17 00:00:00 2001
From: iomgaa <iomgaaycz@gmail.com>
Date: Mon, 12 May 2025 12:24:03 +0800
Subject: [PATCH] update

---
 model/model.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/model/model.py b/model/model.py
index b9fdf14..54d5599 100644
--- a/model/model.py
+++ b/model/model.py
@@ -613,12 +613,15 @@ class MiniMindLM(PreTrainedModel):
             logits=logits,
             past_key_values=past_kvs,
         )
+        output.hidden_states = h
+
+        output.aux_loss = aux_loss
 
         # 尝试添加其他属性（如果支持的话）
-        try:
-            output.hidden_states = h
-        except:
-            pass
+        # try:
+        #     output.hidden_states = h
+        # except:
+        #     pass
 
         return output