Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "f85ab4c83de67a762268540ded699a061374c65f"
Commit 613b4bd4 authored by Baber's avatar Baber
Browse files

add gen kwargs

parent 40d3a138
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
lm_eval/tasks/mrl/gen/_template
View file @ 613b4bd4
...@@ -5,7 +5,9 @@ doc_to_text: "Given the following situation, which option is more likely to be c ...@@ -5,7 +5,9 @@ doc_to_text: "Given the following situation, which option is more likely to be c
doc_to_target: "{{['A', 'B'][label]}}" doc_to_target: "{{['A', 'B'][label]}}"
generation_kwargs: generation_kwargs:
do_sample: true do_sample: true
temperature: 1 temperature: 0.8
top_p: 0.95
repetition_penalty: 1.1
max_gen_toks: 2048 max_gen_toks: 2048
until: [ ] until: [ ]
filter_list: filter_list:
......
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