From 8ed3c3fe2a2fef759069cf13d9bd2eb115cc0bc4 Mon Sep 17 00:00:00 2001
From: xaedes <xaedes@gmail.com>
Date: Fri, 14 Apr 2023 02:38:10 +0200
Subject: [PATCH] reserve correct size for logits

---
 llama.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llama.cpp b/llama.cpp
index 33ee4fbb5..21e4d73fb 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -1787,7 +1787,7 @@ struct llama_context * llama_init_from_file(
         if (params.logits_all) {
             ctx->logits.reserve(hparams.n_ctx*hparams.n_vocab);
         } else {
-            ctx->logits.reserve(hparams.n_ctx);
+            ctx->logits.reserve(hparams.n_vocab);
         }
 
         if (params.embedding){