Skip to content

Commit 3494265

Browse files
Update convert_hf_to_gguf.py
Co-authored-by: compilade <[email protected]>
1 parent 184a4c6 commit 3494265

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2744,7 +2744,7 @@ def set_gguf_parameters(self):
27442744
rms_norm_eps = self.find_hparam(["layer_norm_epsilon", "rms_norm_eps"], optional=True) or 1e-5
27452745
use_b_dt_norm = False
27462746
# For falconmamba we do apply RMS norm on B / DT and C layers
2747-
if self.find_hparam(["model_type"]) in ["falcon_mamba"]:
2747+
if self.find_hparam(["model_type"], optional=True) in ("falcon_mamba",):
27482748
use_b_dt_norm = True
27492749
# Fail early for models which don't have a block expansion factor of 2
27502750
assert d_inner == 2 * d_model

0 commit comments

Comments
 (0)