ZwwWayne commited on
Commit
405ebfe
1 Parent(s): d64cff5

fix(modeling): fix inference code

Browse files
Files changed (1) hide show
  1. 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
+ )