From f7882e2d6943e3ddadcbd287d8f181c10276afdb Mon Sep 17 00:00:00 2001 From: digiwombat Date: Wed, 31 May 2023 20:35:28 -0400 Subject: [PATCH] Fixed a crash caused by erasing from empty last_n_tokens --- examples/server/server.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/server/server.cpp b/examples/server/server.cpp index c12a84fa7..3e27a7bbb 100644 --- a/examples/server/server.cpp +++ b/examples/server/server.cpp @@ -251,7 +251,10 @@ struct llama_server_context id = llama_sample_token(ctx, &candidates_p); } } - last_n_tokens.erase(last_n_tokens.begin()); + if (!last_n_tokens.empty()) + { + last_n_tokens.erase(last_n_tokens.begin()); + } last_n_tokens.push_back(id); num_tokens_predicted++; }