From 5fd28b3aca07b2798fcfd122c677432b93ee0836 Mon Sep 17 00:00:00 2001 From: ochafik Date: Sun, 2 Feb 2025 02:07:32 +0000 Subject: [PATCH] use set::find --- examples/server/server.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/server/server.cpp b/examples/server/server.cpp index 93ecd1502..8a5ef5045 100644 --- a/examples/server/server.cpp +++ b/examples/server/server.cpp @@ -2855,8 +2855,7 @@ struct server_context { server_slot * slot_batched = nullptr; auto accept_special_token = [&](server_slot & slot, llama_token token) { - const auto & preserved_tokens = slot.params.sampling.preserved_tokens; - return params_base.special || std::find(preserved_tokens.begin(), preserved_tokens.end(), token) != preserved_tokens.end(); + return params_base.special || slot.params.sampling.preserved_tokens.find(token) != slot.params.sampling.preserved_tokens.end(); }; // frist, add sampled tokens from any ongoing sequences