Whitespace

This commit is contained in:
S 2024-04-05 10:37:25 +01:00
parent 553b09ba8f
commit f3532ff80c

View file

@ -2344,14 +2344,14 @@ class CommandR2Model(Model):
# max_position_embeddings = 8192 in config.json but model was actually # max_position_embeddings = 8192 in config.json but model was actually
# trained on 128k context length # trained on 128k context length
self.hparams["max_position_embeddings"] = self.hparams["model_max_length"] self.hparams["max_position_embeddings"] = self.hparams["model_max_length"]
def set_gguf_parameters(self): def set_gguf_parameters(self):
super().set_gguf_parameters() super().set_gguf_parameters()
self.gguf_writer.add_logit_scale(self.hparams["logit_scale"]) self.gguf_writer.add_logit_scale(self.hparams["logit_scale"])
self.gguf_writer.add_rope_scaling_type(gguf.RopeScalingType.NONE) self.gguf_writer.add_rope_scaling_type(gguf.RopeScalingType.NONE)
###### CONVERSION LOGIC ###### ###### CONVERSION LOGIC ######