nit: add comment

artemorloff · Aug 29, 2024 · 1c2def6 · 1c2def6
1 parent f2f7fa8
commit 1c2def6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lm_eval/models/vllm_causallms.py b/lm_eval/models/vllm_causallms.py
@@ -290,6 +290,7 @@ def loglikelihood_rolling(
                     get_rolling_token_windows(
                         token_list=self.tok_encode(string),
                         prefix_token=self.prefix_token_id,
+                        # max_seq_len - (1 for context)
                         max_seq_len=self.max_length - 1,
                         context_len=1,
                     ),