summaryrefslogtreecommitdiff
path: root/examples/simple
diff options
context:
space:
mode:
Diffstat (limited to 'examples/simple')
-rw-r--r--examples/simple/simple.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/simple/simple.cpp b/examples/simple/simple.cpp
index 24fb16b7..55385f56 100644
--- a/examples/simple/simple.cpp
+++ b/examples/simple/simple.cpp
@@ -92,7 +92,7 @@ int main(int argc, char ** argv) {
// create a llama_batch with size 512
// we use this object to submit token data for decoding
- llama_batch batch = llama_batch_init(512, 0);
+ llama_batch batch = llama_batch_init(512, 0, 1);
// evaluate the initial prompt
batch.n_tokens = tokens_list.size();