[BugFix] fix RMSNorm rms_norm_esp (#2804)

This commit is contained in:
lizexu123
2025-07-10 20:39:02 +08:00
committed by GitHub
parent 823a47e64a
commit e681e1e719
7 changed files with 18 additions and 16 deletions

View File

@@ -84,6 +84,7 @@ class ModelConfig(PretrainedConfig):
head_dim: Optional[int] = None,
tie_word_embeddings: bool = False,
is_quantized: bool = False,
rms_norm_eps: float = 1e-5,
**kwargs,
):
super().__init__(**kwargs)
@@ -123,6 +124,7 @@ class ModelConfig(PretrainedConfig):
self.dtype = dtype
self.tie_word_embeddings = tie_word_embeddings
self.is_quantized = is_quantized
self.rms_norm_eps = rms_norm_eps
@dataclass