From 2ecb8953b03f3281373727cada299d71dc4d70b1 Mon Sep 17 00:00:00 2001
From: ai-modelscope <ai@modelscope.cn>
Date: Tue, 11 Feb 2025 21:55:14 +0800
Subject: [PATCH] Update modeling_internlm3.py (#18)

- Update modeling_internlm3.py (94cd46f35e87e1b3b2b82df73230bdb5275cd652)
- Update tokenization_internlm3.py (0f3d7019880c0b6f7a9d35b392d21cbfca07478b)
---
 modeling_internlm3.py     | 3 ++-
 tokenization_internlm3.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/modeling_internlm3.py b/modeling_internlm3.py
index 6e877d0..eec4e6b 100644
--- a/modeling_internlm3.py
+++ b/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"}
 
diff --git a/tokenization_internlm3.py b/tokenization_internlm3.py
index f68147f..fb919ad 100644
--- a/tokenization_internlm3.py
+++ b/tokenization_internlm3.py
@@ -67,7 +67,7 @@ class InternLM3Tokenizer(PreTrainedTokenizer):
             Whether or not to add an initial space to the input. This allows to treat the leading word just as any
             other word. Again, this should be set with `from_slow=True` to make sure it's taken into account.
     """
-
+    _auto_class = "AutoTokenizer"
     vocab_files_names = VOCAB_FILES_NAMES
     model_input_names = ["input_ids", "attention_mask"]