drop _tied_weights_keys

mayank31398 · mayank31398 · commit a4e81d3c4a62 · 2026-01-29T23:12:35.000-08:00
Signed-off-by: Mayank Mishra &lt;mayank31398@gmail.com&gt;
diff --git a/lm_engine/hf_models/mixins/dense/base.py b/lm_engine/hf_models/mixins/dense/base.py
@@ -333,7 +333,6 @@ def _setup_positional_encoding(self) -> None:
                     base=self.config.rope_theta,
                 )
             else:
-                print(self.config)
                 self.rope = YaRNScaledRoPE(
                     self.rope_dim,
                     max_position_embeddings=max_position_embeddings,
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,7 +21,7 @@ dependencies = [
     "ninja>=1.13.0",
     "numpy>=2.3.5",
     "pydantic>=2.12.4",
-    "transformers>=4.57.3",
+    "transformers==4.57.3",
     "wandb>=0.23.0",
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -333,7 +333,6 @@ def _setup_positional_encoding(self) -> None:`
`333`	`333`	`base=self.config.rope_theta,`
`334`	`334`	`)`
`335`	`335`	`else:`
`336`		`- print(self.config)`
`337`	`336`	`self.rope = YaRNScaledRoPE(`
`338`	`337`	`self.rope_dim,`
`339`	`338`	`max_position_embeddings=max_position_embeddings,`