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
     ///