Update modeling_internlm2.py for transformers >=4.50.0

modeling_internlm2.py

@@ -34,6 +34,7 @@ from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import (add_start_docstrings,
                                 add_start_docstrings_to_model_forward, logging,
                                 replace_return_docstrings)
+from transformers.generation.utils import GenerationMixin
 
 try:
     from transformers.generation.streamers import BaseStreamer
@@ -1026,7 +1027,7 @@ class InternLM2Model(InternLM2PreTrainedModel):
 
 
 # Modified from transformers.model.llama.modeling_llama.LlamaForCausalLM
-class InternLM2ForCausalLM(InternLM2PreTrainedModel):
+class InternLM2ForCausalLM(InternLM2PreTrainedModel, GenerationMixin):
     _auto_class = 'AutoModelForCausalLM'
 
     _tied_weights_keys = ['output.weight']