vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pr review fixes

Browse source
This commit is contained in:
Michael Podvitskiy 2024-03-13 11:44:08 +01:00
parent 80f66a8af7
commit 0a1322acbd
1 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
convert.py
View file

@ -948,18 +948,14 @@ def bounded_parallel_map(func: Callable[[In], Out], iterable: Iterable[In], conc
yield result yield result
def check_vocab_size(params: Params, vocab: Vocab) -> None: def prepare_vocab(params: Params, vocab: Vocab, pad_vocab: bool = False) -> None:
# Handle special case where the model's vocab size is not set # Handle special case where the model's vocab size is not set
if params.n_vocab == -1: if params.n_vocab == -1:
raise ValueError( raise ValueError(
f"The model's vocab size is set to -1 in params.json. Please update it manually.{f' Maybe {vocab.vocab_size}?' if hasattr(vocab, 'vocab_size') else ''}" f"The model's vocab size is set to -1 in params.json. Please update it manually.{f' Maybe {vocab.vocab_size}?' if hasattr(vocab, 'vocab_size') else ''}"
) )
if isinstance(vocab, NoVocab):
return # model has no vocab
def prepare_vocab(params: Params, vocab: Vocab, pad_vocab: bool = False) -> None:
check_vocab_size(params, vocab)
if vocab.name == "no_vocab":
return
# Check for a vocab size mismatch # Check for a vocab size mismatch
if params.n_vocab == vocab.vocab_size: if params.n_vocab == vocab.vocab_size:
@ -1147,7 +1143,7 @@ class OutputFile:
# meta data # meta data
of.add_meta_arch(params) of.add_meta_arch(params)
if vocab.name == "no_vocab": if isinstance(vocab, NoVocab):
of.gguf.add_tokenizer_model(vocab.tokenizer_model) of.gguf.add_tokenizer_model(vocab.tokenizer_model)
else: else:
of.add_meta_vocab(vocab) of.add_meta_vocab(vocab)