llama: correct vocab size for logging

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
Michael Podvitskiy 2024-09-17 11:23:52 +02:00 committed by GitHub
parent 9704f0e928
commit 93ef595b4b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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;
}