update

2025-05-12 12:16:35 +08:00 · 2025-05-12 12:16:35 +08:00 · 77cf24c925
commit 77cf24c925
parent 6eaca41018
1 changed files with 8 additions and 5 deletions
--- a/model/model.py
+++ b/model/model.py
@ -612,14 +612,17 @@ class MiniMindLM(PreTrainedModel):
            logits=logits,
            past_key_values=past_kvs,
        )
+        output.hidden_states = h
+
+        output.aux_loss = aux_loss

        # 尝试添加其他属性（如果支持的话）
-        try:
-            output.hidden_states = h
+        # try:
+        #     output.hidden_states = h

-            output.aux_loss = aux_loss                                                                        
-        except:
-            pass
+        #     output.aux_loss = aux_loss                                                                        
+        # except:
+        #     pass

        return output