Update modeling_chatglm.py

modeling_chatglm.py

@@ -743,7 +743,7 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
 
         self.embed_tokens = nn.Embedding(
             config.padded_vocab_size,
-            self.hidden_size,
+            config.hidden_size,
             dtype=config.torch_dtype,
             device=device
         )
@@ -825,7 +825,7 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
         batch_size, seq_length = input_ids.shape
 
         if inputs_embeds is None:
-            inputs_embeds = self.embedding(input_ids)
+            inputs_embeds = self.embed_tokens(input_ids)
 
         if self.pre_seq_len is not None:
             if past_key_values is None: