Fix: State-setting assert fails when ctx->logits_all is false

This commit is contained in:
Paul Tsochantaris 2023-12-12 19:02:42 +00:00
parent fecac45658
commit 19dcce5c35

View file

@ -9594,7 +9594,7 @@ size_t llama_set_state_data(struct llama_context * ctx, uint8_t * src) {
memcpy(&logits_cap, inp, sizeof(logits_cap)); inp += sizeof(logits_cap);
memcpy(&logits_size, inp, sizeof(logits_size)); inp += sizeof(logits_size);
GGML_ASSERT(ctx->logits.capacity() == logits_cap);
GGML_ASSERT((!ctx->logits_all) || (ctx->logits.capacity() == logits_cap));
if (logits_size) {
ctx->logits.resize(logits_size);