llama : fix --mtest option (close #1414)

ggerganov · May 12, 2023 · fb62f92 · fb62f92
1 parent 773ee24
commit fb62f92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);
         }