Revert "fix"

This reverts commit 243e4d1a50.
This commit is contained in:
ibrahimkhadraoui
2025-07-04 15:04:13 +04:00
parent cce35498d5
commit 2fe057cc40

View File

@@ -6606,8 +6606,6 @@ class FalconH1Model(Mamba2Model):
for name, tensor in super().get_tensors(): for name, tensor in super().get_tensors():
if name.startswith("model.backbone") or name.startswith("model.lm_head"): if name.startswith("model.backbone") or name.startswith("model.lm_head"):
name = name.removeprefix("model.") name = name.removeprefix("model.")
if "ffn_pre_norm" in name:
name = name.replace("ffn_pre_norm", "ffn_norm")
yield name, tensor yield name, tensor
if self.ssm_multipliers is not None: if self.ssm_multipliers is not None: