diff --git a/examples/mistral/config.yml b/examples/mistral/config.yml index 67a663a62..8edaa06b0 100644 --- a/examples/mistral/config.yml +++ b/examples/mistral/config.yml @@ -47,7 +47,7 @@ flash_attention: true warmup_steps: 10 eval_steps: 20 -eval_table_size: 5 +eval_table_size: eval_table_max_new_tokens: 128 save_steps: debug: diff --git a/examples/mistral/qlora.yml b/examples/mistral/qlora.yml index 4312a0265..ac4bcbd16 100644 --- a/examples/mistral/qlora.yml +++ b/examples/mistral/qlora.yml @@ -64,7 +64,7 @@ flash_attention: true warmup_steps: 10 eval_steps: 20 -eval_table_size: 5 +eval_table_size: eval_table_max_new_tokens: 128 save_steps: debug: