small update

modeling_internlm2.py

@@ -1632,10 +1632,10 @@ class InternLM2ForQuestionAnswering(InternLM2PreTrainedModel):
         self.post_init()
 
     def get_input_embeddings(self):
-        return self.transformer.embed_tokens
+        return self.transformer.tok_embeddings
 
     def set_input_embeddings(self, value):
-        self.transformer.embed_tokens = value
+        self.transformer.tok_embeddings = value
 
     @add_start_docstrings_to_model_forward(InternLM2_INPUTS_DOCSTRING)
     def forward(
@@ -1739,10 +1739,10 @@ class InternLM2ForTokenClassification(InternLM2PreTrainedModel):
         self.post_init()
 
     def get_input_embeddings(self):
-        return self.model.embed_tokens
+        return self.model.tok_embeddings
 
     def set_input_embeddings(self, value):
-        self.model.embed_tokens = value
+        self.model.tok_embeddings = value
 
     @add_start_docstrings_to_model_forward(InternLM2_INPUTS_DOCSTRING)
     def forward(