Update modeling_internlm2.py

modeling_internlm2.py

@@ -865,7 +865,7 @@ class InternLM2Model(InternLM2PreTrainedModel):
 
         seq_length_with_past = seq_length
         past_key_values_length = 0
-        if past_key_values is not None:
+        if past_key_values is not None and past_key_values[0][0] is not None:
             past_key_values_length = past_key_values[0][0].shape[2]
             seq_length_with_past = seq_length_with_past + past_key_values_length