Skip to content

Commit 35782ae

Browse files
authored
convert : BailingMoE : avoid setting rope_dim to 0 (ggml-org#12678)
1 parent c80a775 commit 35782ae

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
@@ -5146,7 +5146,7 @@ def set_vocab(self):
51465146
def set_gguf_parameters(self):
51475147
super().set_gguf_parameters()
51485148
hparams = self.hparams
5149-
if "head_dim" in hparams:
5149+
if hparams.get("head_dim"):
51505150
rope_dim = hparams["head_dim"]
51515151
else:
51525152
rope_dim = hparams["hidden_size"] // hparams["num_attention_heads"]

0 commit comments

Comments
 (0)