convert : fix invalid params in write_vocab_only (ggerganov#1975)

christianazinn · Jun 24, 2023 · c943d82 · c943d82
1 parent f2c754e
commit c943d82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/convert.py b/convert.py
@@ -998,9 +998,9 @@ def write_vocab(self, vocab: Vocab) -> None:
     def write_vocab_only(fname_out: Path, vocab: Vocab) -> None:
         of = OutputFile(fname_out)
         params = Params(n_vocab=vocab.vocab_size, n_embd=0, n_mult=0,
-                        n_head=1, n_layer=0, file_type=GGMLFileType.AllF32)
+                        n_head=1, n_layer=0)
         of = OutputFile(fname_out)
-        of.write_file_header(params)
+        of.write_file_header(params, file_type=GGMLFileType.AllF32)
         of.write_vocab(vocab)
         of.fout.close()