Просмотр исходного кода

convert : fix missing ftype for gemma (#5690)

Jared Van Bortel 1 год назад
Родитель
Сommit
54fbcd2ce6
1 измененных файлов с 1 добавлено и 0 удалено
  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")))