Skip to content

Commit 5fd28b3

Browse files
author
ochafik
committed
use set::find
1 parent a28d9be commit 5fd28b3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/server/server.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2855,8 +2855,7 @@ struct server_context {
28552855
server_slot * slot_batched = nullptr;
28562856

28572857
auto accept_special_token = [&](server_slot & slot, llama_token token) {
2858-
const auto & preserved_tokens = slot.params.sampling.preserved_tokens;
2859-
return params_base.special || std::find(preserved_tokens.begin(), preserved_tokens.end(), token) != preserved_tokens.end();
2858+
return params_base.special || slot.params.sampling.preserved_tokens.find(token) != slot.params.sampling.preserved_tokens.end();
28602859
};
28612860

28622861
// frist, add sampled tokens from any ongoing sequences

0 commit comments

Comments
 (0)