fix max_gen_toks

d985629d · Baber · f4b4f486 · d985629d
Commit d985629d authored Jul 30, 2025 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/tasks/mbpp/mbpp_evalplus.yaml lm_eval/tasks/mbpp/mbpp_evalplus.yaml +1 -1

No files found.
--- a/lm_eval/tasks/mbpp/mbpp_evalplus.yaml
+++ b/lm_eval/tasks/mbpp/mbpp_evalplus.yaml
@@ -40,7 +40,7 @@ generation_kwargs:
  do_sample: true
  temperature: 0.8
  top_p: 0.95
-  max_gen_toks: 2
+  max_gen_toks: 512
 num_fewshot: 0
 fewshot_config:
  sampler: first_n