[HF] drop _tied_weights_keys (#363)

Signed-off-by: Mayank Mishra <mayank31398@gmail.com>

Author: mayank31398

lm_engine/hf_models/mixins/dense/main.py

@@ -19,7 +19,6 @@
 
 
 class CausalLMModelMixin(PreTrainedModelMixin):
-    _tied_weights_keys = ["lm_head.weight"]
     base_model_class = None
 
     def __init__(self, config: CommonConfig, **kwargs) -> CausalLMModelMixin:

pyproject.toml

@@ -21,7 +21,7 @@ dependencies = [
     "ninja>=1.13.0",
     "numpy>=2.3.5",
     "pydantic>=2.12.4",
-    "transformers>=4.57.1",
+    "transformers==4.57.3",
     "wandb>=0.23.0",
 ]