diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 37d6cc2627c13..3cf8b5165fa98 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -5146,7 +5146,7 @@ def set_vocab(self): def set_gguf_parameters(self): super().set_gguf_parameters() hparams = self.hparams - if "head_dim" in hparams: + if hparams.get("head_dim"): rope_dim = hparams["head_dim"] else: rope_dim = hparams["hidden_size"] // hparams["num_attention_heads"]