Explorar o código

convert : BailingMoE : avoid setting rope_dim to 0 (#12678)

Sigbjørn Skjæret hai 9 meses
pai
achega
35782aeedb
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      convert_hf_to_gguf.py

+ 1 - 1
convert_hf_to_gguf.py

@@ -5146,7 +5146,7 @@ class BailingMoeModel(Model):
     def set_gguf_parameters(self):
     def set_gguf_parameters(self):
         super().set_gguf_parameters()
         super().set_gguf_parameters()
         hparams = self.hparams
         hparams = self.hparams
-        if "head_dim" in hparams:
+        if hparams.get("head_dim"):
             rope_dim = hparams["head_dim"]
             rope_dim = hparams["head_dim"]
         else:
         else:
             rope_dim = hparams["hidden_size"] // hparams["num_attention_heads"]
             rope_dim = hparams["hidden_size"] // hparams["num_attention_heads"]