diff options
author | AN Long <aisk@users.noreply.github.com> | 2023-06-24 19:02:06 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-06-24 14:02:06 +0300 |
commit | c943d823c14cef33092205ca3944de6fdf7abf99 (patch) | |
tree | 37b0f1723e729d017157d60bb2fcba4317f7d70c | |
parent | f2c754e1c38936fdde74e4848ac468a696eb73c6 (diff) |
convert : fix invalid params in write_vocab_only (#1975)
-rw-r--r-- | convert.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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() |