Skip to content

Commit 132f557

Browse files
authored
llama : fix restoring the number of outputs from state files (#6687)
1 parent 3272896 commit 132f557

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

llama.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15478,6 +15478,8 @@ size_t llama_state_set_data(struct llama_context * ctx, const uint8_t * src) {
1547815478
GGML_ASSERT((uint32_t) id < ctx->cparams.n_batch);
1547915479
ctx->output_ids[id] = i;
1548015480
}
15481+
15482+
ctx->n_outputs = n_outputs;
1548115483
}
1548215484
}
1548315485

0 commit comments

Comments
 (0)