convert : fix missing ftype for gemma (ggml-org#5690)

hodlen · Feb 23, 2024 · 54fbcd2 · 54fbcd2
1 parent 15499eb
commit 54fbcd2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -1803,6 +1803,7 @@ def set_gguf_parameters(self):
         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")))