]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
convert : fix invalid params in write_vocab_only (#1975)
authorAN Long <redacted>
Sat, 24 Jun 2023 11:02:06 +0000 (19:02 +0800)
committerGitHub <redacted>
Sat, 24 Jun 2023 11:02:06 +0000 (14:02 +0300)
convert.py

index de6c39c67672bf77bae8aae698e449632d9cf2c4..e340d2273f378ab0d81ef3ba20e23408c30a78dd 100644 (file)
@@ -998,9 +998,9 @@ class OutputFile:
     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()