aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2023-05-12 21:44:20 +0300
committerGeorgi Gerganov <ggerganov@gmail.com>2023-05-12 21:44:20 +0300
commitfb62f924336c9746da9976c6ab3c2e6460258d54 (patch)
tree6922de287ac7c284b77dbaf48f1f203b87d23cac
parent773ee249fb6c14f791ac39f6ec05536f40dedc54 (diff)
llama : fix --mtest option (close #1414)
-rw-r--r--examples/main/main.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index bd1c4ab..8543414 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -121,7 +121,7 @@ int main(int argc, char ** argv) {
// uncomment the "used_mem" line in llama.cpp to see the results
if (params.mem_test) {
{
- const std::vector<llama_token> tmp(params.n_batch, 0);
+ const std::vector<llama_token> tmp(params.n_batch, llama_token_bos());
llama_eval(ctx, tmp.data(), tmp.size(), 0, params.n_threads);
}