diff --git a/examples/beam_search/beam_search.cpp b/examples/beam_search/beam_search.cpp index 95a8e5179..29b69b76d 100644 --- a/examples/beam_search/beam_search.cpp +++ b/examples/beam_search/beam_search.cpp @@ -34,7 +34,7 @@ struct ostream_beam_view { }; std::ostream& operator<<(std::ostream& os, ostream_beam_view const & obv) { os << "p(" << obv.beam_view.p << ") eos(" << std::boolalpha << obv.beam_view.eos << ") tokens("; - for (size_t i=0 ; i(callback_data_ptr); // Mark beams as EOS as needed. - for (size_t i=0 ; i(callback_data); // Mark beams as EOS as needed. - for (size_t i=0 ; i min_heap; // min-heap by logit llama_token const k_min = std::min(static_cast(k), n_vocab); min_heap.reserve(k_min); - for (llama_token token_id=0 ; token_id b.logit; }; std::make_heap(min_heap.begin(), min_heap.end(), comp); - for (llama_token token_id=k_min ; token_id