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
|
||||
auto & logits_valid = lctx.logits_valid;
|
||||
logits_valid.clear();
|
||||
logits_valid.resize(n_vocab);
|
||||
logits_valid.resize(n_tokens);
|
||||
#endif
|
||||
|
||||
if (batch.logits) {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue