Update modeling_chatglm.py

modeling_chatglm.py

@@ -705,6 +705,9 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
     def get_input_embeddings(self):
         return self.embedding.word_embeddings
 
+    def set_input_embeddings(self, new_embeddings: torch.Tensor):
+        self.word_embeddings = new_embeddings
+    
     def forward(
             self,
             input_ids,