diff options
author | Galunid <karolek1231456@gmail.com> | 2024-05-26 16:02:34 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-26 16:02:34 +0200 |
commit | 32a28217f475119926c603341e8273b26932b56a (patch) | |
tree | d1018dfd246cebf2864b6c9f9de4c3742f5960db | |
parent | c429b33beb35f13934a4dfbe0c138d30b45e5d54 (diff) |
Fix aya-23 conversion scripts (#7539)
-rwxr-xr-x | convert-hf-to-gguf.py | 3 |
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() |