ZwwWayne
commited on
Commit
•
405ebfe
1
Parent(s):
d64cff5
fix(modeling): fix inference code
Browse files- modeling_internlm2.py +2 -1
modeling_internlm2.py
CHANGED
@@ -1268,4 +1268,5 @@ class InternLM2ForSequenceClassification(InternLM2PreTrainedModel):
|
|
1268 |
logits=pooled_logits,
|
1269 |
past_key_values=transformer_outputs.past_key_values,
|
1270 |
hidden_states=transformer_outputs.hidden_states,
|
1271 |
-
attentions=transformer_outputs
|
|
|
|
1268 |
logits=pooled_logits,
|
1269 |
past_key_values=transformer_outputs.past_key_values,
|
1270 |
hidden_states=transformer_outputs.hidden_states,
|
1271 |
+
attentions=transformer_outputs,
|
1272 |
+
)
|