nit

55020c98 · Baber · 6e7c7897 · 55020c98
Commit 55020c98 authored Feb 20, 2025 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 6 deletions

lm_eval/tasks/gsm8k/gsm8k-cot-llama.yaml lm_eval/tasks/gsm8k/gsm8k-cot-llama.yaml +2 -6

No files found.
--- a/lm_eval/tasks/gsm8k/gsm8k-cot-llama.yaml
+++ b/lm_eval/tasks/gsm8k/gsm8k-cot-llama.yaml
@@ -44,12 +44,8 @@ fewshot_config:
      dollars. So she has 23 - 15 dollars left. 23 - 15 is 8. The final answer is 8
 generation_kwargs:
  do_sample: false
-  until:
-  - '<|eot_id|>'
-  - '<|start_header_id|>user<|end_header_id|>'
-  - 'Q:'
-  - </s>
-  - <|im_end|>
+  until: []
+  max_gen_toks: 1024
 tag:
 - chain_of_thought
 metadata: