llama : add comment about llama_sample_token_greedy() missing probs

2023-10-28 13:21:29 +03:00 · 2023-10-28 13:21:29 +03:00 · c86cca8061
parent 4aa1fb0d38
commit c86cca8061
1 changed files with 1 additions and 0 deletions
--- a/llama.h
+++ b/llama.h
@ -658,6 +658,7 @@ extern "C" {
                           float * mu);

    /// @details Selects the token with the highest probability.
+    ///          Does not compute the token probabilities. Use llama_sample_softmax() instead.
    LLAMA_API llama_token llama_sample_token_greedy(
            struct llama_context * ctx,
          llama_token_data_array * candidates);