update

2025-05-12 12:24:03 +08:00 · 2025-05-12 12:24:03 +08:00 · decec67b78
commit decec67b78
parent d93889194d
1 changed files with 7 additions and 4 deletions
--- 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