diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-12 21:44:20 +0300 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2023-05-12 21:44:20 +0300 |
commit | fb62f924336c9746da9976c6ab3c2e6460258d54 (patch) | |
tree | 6922de287ac7c284b77dbaf48f1f203b87d23cac /examples/main | |
parent | 773ee249fb6c14f791ac39f6ec05536f40dedc54 (diff) |
llama : fix --mtest option (close #1414)
Diffstat (limited to 'examples/main')
-rw-r--r-- | examples/main/main.cpp | 2 |
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); } |