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

fix simple.cpp

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-10-11 12:16:03 +02:00
parent 997031605b
commit 92769503dc
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/simple/simple.cpp
View file

@ -138,7 +138,7 @@ int main(int argc, char ** argv) {
// prepare a batch for the prompt // prepare a batch for the prompt
llama_batch batch = llama_batch_get_one(prompt_tokens.data(), prompt_tokens.size(), 0, 0); llama_batch batch = llama_batch_get_one(prompt_tokens.data(), prompt_tokens.size());
// main loop // main loop
@ -175,7 +175,7 @@ int main(int argc, char ** argv) {
fflush(stdout); fflush(stdout);
// prepare the next batch with the sampled token // prepare the next batch with the sampled token
batch = llama_batch_get_one(&new_token_id, 1, n_pos, 0); batch = llama_batch_get_one(&new_token_id, 1);
n_decode += 1; n_decode += 1;
} }