Use a set's `find` method when looking for reject tokens.

PiperOrigin-RevId: 731332151
2025-02-26 08:42:36 -08:00 · 2025-02-26 08:42:36 -08:00 · de5bab65b4
parent 2bdf26d81d
commit de5bab65b4
1 changed files with 1 additions and 1 deletions
--- a/examples/hello_world/run.cc
+++ b/examples/hello_world/run.cc
@ -99,7 +99,7 @@ int main(int argc, char** argv) {
      .stream_token = stream_token,
      .accept_token =
          [&](int token, float /* prob */) {
-            return !reject_tokens.contains(token);
+            return reject_tokens.find(token) == reject_tokens.end();
          },
  };
  model.Generate(runtime_config, tokens, 0, kv_cache, timing_info);