llama: correct vocab size for logging

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Xarbirus · Sep 17, 2024 · 93ef595 · 93ef595
1 parent 9704f0e
commit 93ef595
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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;
         }