n_vocab -> n_tokens
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
parent
f91707bbe1
commit
5284e72aa6
1 changed files with 1 additions and 1 deletions
|
@ -5616,7 +5616,7 @@ static int llama_decode_internal(
|
||||||
#ifndef NDEBUG
|
#ifndef NDEBUG
|
||||||
auto & logits_valid = lctx.logits_valid;
|
auto & logits_valid = lctx.logits_valid;
|
||||||
logits_valid.clear();
|
logits_valid.clear();
|
||||||
logits_valid.resize(n_vocab);
|
logits_valid.resize(n_tokens);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (batch.logits) {
|
if (batch.logits) {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue