diff --git a/examples/llama-2/lora.yml b/examples/llama-2/lora.yml index 3134648fdb..106de6e320 100644 --- a/examples/llama-2/lora.yml +++ b/examples/llama-2/lora.yml @@ -56,7 +56,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/llama-2/qlora.yml b/examples/llama-2/qlora.yml index 54255320cc..870c3fa848 100644 --- a/examples/llama-2/qlora.yml +++ b/examples/llama-2/qlora.yml @@ -58,7 +58,7 @@ flash_attention: true warmup_steps: 10 eval_steps: 20 -eval_table_size: 5 +eval_table_size: save_steps: debug: deepspeed: diff --git a/examples/llama-2/tiny-llama.yml b/examples/llama-2/tiny-llama.yml index a53c9c831b..d556a6d719 100644 --- a/examples/llama-2/tiny-llama.yml +++ b/examples/llama-2/tiny-llama.yml @@ -56,7 +56,7 @@ flash_attention: true warmup_steps: 10 eval_steps: 20 -eval_table_size: 5 +eval_table_size: save_steps: debug: deepspeed: