llama: correct vocab size for logging
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
parent
9704f0e928
commit
93ef595b4b
1 changed files with 2 additions and 1 deletions
|
@ -6063,7 +6063,8 @@ static void llm_load_vocab(
|
|||
|
||||
// read vocab size from metadata
|
||||
if (!ml.get_key(LLM_KV_VOCAB_SIZE, vocab.n_vocab, false)) {
|
||||
LLAMA_LOG_WARN("%s: there is no vocab_size in metadata, vocab.n_vocab will be set to 0\n", __func__);
|
||||
vocab.n_vocab = 0;
|
||||
LLAMA_LOG_WARN("%s: there is no vocab_size in metadata, vocab.n_vocab will be set to %u\n", __func__, vocab.n_vocab);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue