Fix aya-23 conversion scripts (#7539)

This commit is contained in:
Galunid 2024-05-26 16:02:34 +02:00 committed by GitHub
parent c429b33beb
commit 32a28217f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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()