Update modeling_neollm.py

modeling_neollm.py

@@ -1210,10 +1210,14 @@ class NeoLLMForCausalLM(NeoLLMPreTrainedModel, GenerationMixin):
         self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
         
         self.post_init()
-    def tie_weights(self):
+    def tie_weights(self, missing_keys=None, recompute_mapping=True):
         """
         Tie the weights between the input embeddings and the output embeddings.
         Required for v5.0 compatibility.
+        
+        Args:
+            missing_keys: List of missing keys from checkpoint loading (v5.0 parameter)
+            recompute_mapping: Whether to recompute tied weights mapping (v5.0 parameter)
         """
         self._tie_or_clone_weights(self.lm_head, self.model.embed_tokens)
     def forward(