Skip to content

Commit c943d82

Browse files
authored
convert : fix invalid params in write_vocab_only (#1975)
1 parent f2c754e commit c943d82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

convert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -998,9 +998,9 @@ def write_vocab(self, vocab: Vocab) -> None:
998998
def write_vocab_only(fname_out: Path, vocab: Vocab) -> None:
999999
of = OutputFile(fname_out)
10001000
params = Params(n_vocab=vocab.vocab_size, n_embd=0, n_mult=0,
1001-
n_head=1, n_layer=0, file_type=GGMLFileType.AllF32)
1001+
n_head=1, n_layer=0)
10021002
of = OutputFile(fname_out)
1003-
of.write_file_header(params)
1003+
of.write_file_header(params, file_type=GGMLFileType.AllF32)
10041004
of.write_vocab(vocab)
10051005
of.fout.close()
10061006

0 commit comments

Comments
 (0)