vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

reserve correct size for logits

Browse source
This commit is contained in:
xaedes 2023-04-14 02:38:10 +02:00
parent 3d59769c3b
commit 8ed3c3fe2a
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama.cpp
View file

@ -1787,7 +1787,7 @@ struct llama_context * llama_init_from_file(
if (params.logits_all) { if (params.logits_all) {
ctx->logits.reserve(hparams.n_ctx*hparams.n_vocab); ctx->logits.reserve(hparams.n_ctx*hparams.n_vocab);
} else { } else {
ctx->logits.reserve(hparams.n_ctx); ctx->logits.reserve(hparams.n_vocab);
} }
if (params.embedding){ if (params.embedding){