Fix naming

7dd1ef95 · jon-tow · eda365f6 · 7dd1ef95
Commit 7dd1ef95 authored Apr 29, 2022 by jon-tow
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/base.py lm_eval/base.py +1 -1

No files found.
--- a/lm_eval/base.py
+++ b/lm_eval/base.py
@@ -375,7 +375,7 @@ class BaseLM(LM):
            ).to(self.device)

            if max_generation_length is None:
-                max_length = self.max_gen_tok
+                max_length = self.max_gen_toks
            else:
                max_length = max_generation_length