From 4c2cd79b3f801662ff464e1d09b42ffd6d5eb924 Mon Sep 17 00:00:00 2001 From: marcus Date: Thu, 7 Mar 2024 11:32:33 -0800 Subject: [PATCH] added notes on what each number is centered around --- llama-cpp-2/src/token/data_array.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama-cpp-2/src/token/data_array.rs b/llama-cpp-2/src/token/data_array.rs index dc4d0643..1e535d7c 100644 --- a/llama-cpp-2/src/token/data_array.rs +++ b/llama-cpp-2/src/token/data_array.rs @@ -93,9 +93,9 @@ impl LlamaTokenDataArray { /// /// * `ctx` - the context to use. May be `None` if you do not care to record the sample timings. /// * `last_tokens` - the last tokens in the context. - /// * `penalty_last_n` - the number of tokens to consider for the repetition penalty. - /// * `penalty_repeat` - the repetition penalty. - /// * `penalty_freq` - the frequency penalty. + /// * `penalty_last_n` - the number of tokens to consider for the repetition penalty. (1.0 for no penalty) + /// * `penalty_repeat` - the repetition penalty. (0.0 for no penalty) + /// * `penalty_freq` - the frequency penalty. (0.0 for no penalty) /// /// # Example ///