From: compilade <redacted>
Date: Mon, 15 Apr 2024 12:56:55 +0000 (-0400)
Subject: llama : fix restoring the number of outputs from state files (#6687)
X-Git-Tag: upstream/0.0.4488~1809
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=132f55795e51094954f1b1f647f97648be724a3a;p=pkg%2Fggml%2Fsources%2Fllama.cpp

llama : fix restoring the number of outputs from state files (#6687)
---

diff --git a/llama.cpp b/llama.cpp
index cf95cea1..a5ef2fd8 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -15478,6 +15478,8 @@ size_t llama_state_set_data(struct llama_context * ctx, const uint8_t * src) {
                 GGML_ASSERT((uint32_t) id < ctx->cparams.n_batch);
                 ctx->output_ids[id] = i;
             }
+
+            ctx->n_outputs = n_outputs;
         }
     }