Update modeling_neollm.py

modeling_neollm.py

@@ -1210,16 +1210,8 @@ class NeoLLMForCausalLM(NeoLLMPreTrainedModel, GenerationMixin):
         self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
         
         self.post_init()
-    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)
+        _tied_weights_keys = {"lm_head.weight": "model.embed_tokens.weight"}
+
     def forward(
         self,
         input_ids: Optional[torch.LongTensor] = None,