convert : fix missing ftype for gemma (#5690)

This commit is contained in:
Jared Van Bortel 2024-02-23 13:39:14 -05:00 committed by GitHub
parent 15499eb942
commit 54fbcd2ce6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1803,6 +1803,7 @@ class GemmaModel(Model):
self.gguf_writer.add_layer_norm_rms_eps(self.hparams["rms_norm_eps"])
self.gguf_writer.add_key_length(hparams["head_dim"])
self.gguf_writer.add_value_length(hparams["head_dim"])
self.gguf_writer.add_file_type(self.ftype)
def write_tensors(self):
block_count = self.hparams.get("n_layers", self.hparams.get("num_hidden_layers", self.hparams.get("n_layer")))