Sfoglia il codice sorgente

convert : fix missing ftype for gemma (#5690)

Jared Van Bortel 1 anno fa
parent
commit
54fbcd2ce6
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      convert-hf-to-gguf.py

+ 1 - 0
convert-hf-to-gguf.py

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