Skip to content

GitLab

Commit d985629d authored by Baber's avatar Baber
Browse files

fix max_gen_toks

parent f4b4f486
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/tasks/mbpp/mbpp_evalplus.yaml
View file @ d985629d
...@@ -40,7 +40,7 @@ generation_kwargs: ...@@ -40,7 +40,7 @@ generation_kwargs:
do_sample: true do_sample: true
temperature: 0.8 temperature: 0.8
top_p: 0.95 top_p: 0.95
max_gen_toks: 2 max_gen_toks: 512
num_fewshot: 0 num_fewshot: 0
fewshot_config: fewshot_config:
sampler: first_n sampler: first_n
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment