added notes on what each number is centered around

utilityai · Mar 7, 2024 · 4c2cd79 · 4c2cd79
1 parent 7e6a6e7
commit 4c2cd79
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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
     ///