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

passkey : fix llama_kv_cache_seq_pos_max() usage

Browse source
This commit is contained in:
Georgi Gerganov 2024-02-25 10:58:18 +02:00
parent b75ec64ed2
commit 032ff85706
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
examples/passkey/passkey.cpp
View file

@ -150,7 +150,7 @@ int main(int argc, char ** argv) {
llama_kv_cache_seq_div(ctx, 0, n_past - n_batch + ib*bd, n_past + ib*bd, n_grp);
llama_kv_cache_update (ctx);
n_past = llama_kv_cache_seq_pos_max(ctx, 0);
n_past = llama_kv_cache_seq_pos_max(ctx, 0) + 1;
}
llama_batch_clear(batch);
@ -184,7 +184,7 @@ int main(int argc, char ** argv) {
llama_kv_cache_seq_add(ctx, 0, n_keep + n_discard, n_ctx, -n_discard);
llama_kv_cache_update (ctx);
n_past = llama_kv_cache_seq_pos_max(ctx, 0);
n_past = llama_kv_cache_seq_pos_max(ctx, 0) + 1;
llama_batch_clear(batch);
@ -214,7 +214,7 @@ int main(int argc, char ** argv) {
llama_kv_cache_seq_add(ctx, 0, n_keep + n_discard, n_ctx, -n_discard);
llama_kv_cache_update (ctx);
n_past = llama_kv_cache_seq_pos_max(ctx, 0);
n_past = llama_kv_cache_seq_pos_max(ctx, 0) + 1;
}
}