nit

28abec9f · Baber · bf90aa7e · 28abec9f
Commit 28abec9f authored Dec 20, 2024 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 8 deletions

lm_eval/tasks/ruler/prepare.py lm_eval/tasks/ruler/prepare.py +8 -8

No files found.
--- a/lm_eval/tasks/ruler/prepare.py
+++ b/lm_eval/tasks/ruler/prepare.py
@@ -21,14 +21,14 @@ NUM_SAMPLES = 500
 REMOVE_NEWLINE_TAB = ""
 STOP_WORDS = ""
 RANDOM_SEED = 42
-SEQ_LENGTHS = (
-    # 131072,
-    # 65536,
-    # 32768,
-    16384,
-    8192,
-    4096,
-)
+# SEQ_LENGTHS = (
+#     # 131072,
+#     # 65536,
+#     # 32768,
+#     16384,
+#     8192,
+#     4096,
+# )

 # # Define Needle/Haystack Format
 NEEDLE = "One of the special magic {type_needle_v} for {key} is: {value}."