summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGalunid <karolek1231456@gmail.com>2024-05-26 16:02:34 +0200
committerGitHub <noreply@github.com>2024-05-26 16:02:34 +0200
commit32a28217f475119926c603341e8273b26932b56a (patch)
treed1018dfd246cebf2864b6c9f9de4c3742f5960db
parentc429b33beb35f13934a4dfbe0c138d30b45e5d54 (diff)
Fix aya-23 conversion scripts (#7539)
-rwxr-xr-xconvert-hf-to-gguf.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index bfccf862..a342f6b1 100755
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@@ -2395,7 +2395,8 @@ class CommandR2Model(Model):
# max_position_embeddings = 8192 in config.json but model was actually
# trained on 128k context length
- self.hparams["max_position_embeddings"] = self.hparams["model_max_length"]
+ # aya-23 models don't have model_max_length specified
+ self.hparams["max_position_embeddings"] = self.find_hparam(["model_max_length", "max_position_embeddings"])
def set_gguf_parameters(self):
super().set_gguf_parameters()