update

model/model.py

@@ -612,14 +612,17 @@ class MiniMindLM(PreTrainedModel):
             logits=logits,
             past_key_values=past_kvs,
         )
-
-        # 尝试添加其他属性（如果支持的话）
-        try:
         output.hidden_states = h
 
         output.aux_loss = aux_loss
-        except:
-            pass
+
+        # 尝试添加其他属性（如果支持的话）
+        # try:
+        #     output.hidden_states = h
+
+        #     output.aux_loss = aux_loss                                                                        
+        # except:
+        #     pass
 
         return output