Skip to content

GitLab

Commit 83ca7278 authored by Baber's avatar Baber
Browse files

fix max_gen_toks

parent b56ec435
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lm_eval/tasks/ruler/cwe.yaml
View file @ 83ca7278
......@@ -5,5 +5,5 @@ target_delimiter: "\n\n"
generation_kwargs:
do_sample: false
temperature: 0.0
max_gen_toks: 30
max_gen_toks: 120
until: []
lm_eval/tasks/ruler/niah_single_1.yaml
View file @ 83ca7278
......@@ -32,7 +32,7 @@ metric_list:
generation_kwargs:
do_sample: false
temperature: 0.0
max_gen_toks: 64
max_gen_toks: 128
until: []
repeats: 1
metadata:
......
lm_eval/tasks/ruler/vt.yaml
View file @ 83ca7278
......@@ -4,5 +4,5 @@ download_dataset: !function vt_utils.get_vt_dataset
generation_kwargs:
do_sample: false
temperature: 0.0
max_gen_toks: 128
max_gen_toks: 30
until: []
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