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" ...@@ -5,5 +5,5 @@ target_delimiter: "\n\n"
generation_kwargs: generation_kwargs:
do_sample: false do_sample: false
temperature: 0.0 temperature: 0.0
max_gen_toks: 30 max_gen_toks: 120
until: [] until: []
lm_eval/tasks/ruler/niah_single_1.yaml
View file @ 83ca7278
...@@ -32,7 +32,7 @@ metric_list: ...@@ -32,7 +32,7 @@ metric_list:
generation_kwargs: generation_kwargs:
do_sample: false do_sample: false
temperature: 0.0 temperature: 0.0
max_gen_toks: 64 max_gen_toks: 128
until: [] until: []
repeats: 1 repeats: 1
metadata: metadata:
......
lm_eval/tasks/ruler/vt.yaml
View file @ 83ca7278
...@@ -4,5 +4,5 @@ download_dataset: !function vt_utils.get_vt_dataset ...@@ -4,5 +4,5 @@ download_dataset: !function vt_utils.get_vt_dataset
generation_kwargs: generation_kwargs:
do_sample: false do_sample: false
temperature: 0.0 temperature: 0.0
max_gen_toks: 128 max_gen_toks: 30
until: [] 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