Update modeling_internlm3.py

modeling_internlm3.py

@@ -793,7 +793,7 @@ class InternLM3Model(InternLM3PreTrainedModel):
     Args:
         config: InternLM3Config
     """
-
+    _auto_class = "AutoModel"
     def __init__(self, config: InternLM3Config):
         super().__init__(config)
         self.padding_idx = config.pad_token_id
@@ -1070,6 +1070,7 @@ class KwargsForCausalLM(FlashAttentionKwargs, LossKwargs): ...
 
 
 class InternLM3ForCausalLM(InternLM3PreTrainedModel, GenerationMixin):
+    _auto_class = "AutoModelForCausalLM"
     _tied_weights_keys = ["lm_head.weight"]
     _tp_plan = {"lm_head": "colwise_rep"}