Update src/transformers/models/jais2/modular_jais2.py

sarathc-cerebras · vasqu · web-flow · commit 0d2e4b56b967 · 2025-12-13T00:37:15.000+05:30
Co-authored-by: Anton Vlasjuk &lt;73884904+vasqu@users.noreply.github.com&gt;
diff --git a/src/transformers/models/jais2/modular_jais2.py b/src/transformers/models/jais2/modular_jais2.py
@@ -152,7 +152,7 @@ def __init__(
             rope_parameters=rope_parameters,
             **kwargs,
         )
-        # Rename the attribute from rms_norm_eps to layer_norm_eps
+        del self.rms_norm_eps
         self.layer_norm_eps = self.rms_norm_eps
 
         # Validate and standardize RoPE parameters

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,7 @@ def __init__(`
`152`	`152`	`rope_parameters=rope_parameters,`
`153`	`153`	`**kwargs,`
`154`	`154`	`)`
`155`		`- # Rename the attribute from rms_norm_eps to layer_norm_eps`
	`155`	`+ del self.rms_norm_eps`
`156`	`156`	`self.layer_norm_eps = self.rms_norm_eps`
`157`	`157`
`158`	`158`	`# Validate and standardize RoPE parameters`