From 93ef595b4bb7860b9c7438e6cfdf300488ff1380 Mon Sep 17 00:00:00 2001 From: Michael Podvitskiy Date: Tue, 17 Sep 2024 11:23:52 +0200 Subject: [PATCH] llama: correct vocab size for logging Co-authored-by: Georgi Gerganov --- src/llama.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/llama.cpp b/src/llama.cpp index d7855717a..1d8bdbe72 100644 --- a/src/llama.cpp +++ b/src/llama.cpp @@ -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; }