diff --git a/apps/rl/llama3_8b.yaml b/apps/rl/llama3_8b.yaml index d4c291340..58111ebb8 100644 --- a/apps/rl/llama3_8b.yaml +++ b/apps/rl/llama3_8b.yaml @@ -46,7 +46,7 @@ trainer: disable_loss_parallel: false checkpoint: - enable_checkpoint: true + enable: true folder: /tmp/Meta-Llama-3.1-8B-Instruct/saved_checkpoints initial_load_path: /tmp/Meta-Llama-3.1-8B-Instruct/ initial_load_in_hf: true @@ -119,7 +119,7 @@ replay_buffer: # disable_loss_parallel: false # # checkpoint: -# enable_checkpoint: true +# enable: true # folder: /tmp/Meta-Llama-3.1-8B-Instruct/ # interval: 500 # async_mode: "disabled" diff --git a/apps/sft/llama3_8b.yaml b/apps/sft/llama3_8b.yaml index 567e4bad1..3d67cdb23 100644 --- a/apps/sft/llama3_8b.yaml +++ b/apps/sft/llama3_8b.yaml @@ -55,7 +55,7 @@ parallelism: disable_loss_parallel: false checkpoint: - enable_checkpoint: true + enable: true folder: /tmp/Meta-Llama-3.1-8B-Instruct/saved_checkpoints initial_load_path: /tmp/Meta-Llama-3.1-8B-Instruct/ initial_load_in_hf: true diff --git a/apps/sft_v2/llama3_8b.yaml b/apps/sft_v2/llama3_8b.yaml index ad3c0eccf..b033c7626 100644 --- a/apps/sft_v2/llama3_8b.yaml +++ b/apps/sft_v2/llama3_8b.yaml @@ -47,7 +47,7 @@ parallelism: disable_loss_parallel: false checkpoint: - enable_checkpoint: true + enable: true folder: /tmp/Meta-Llama-3.1-8B-Instruct/saved_checkpoints initial_load_path: /tmp/Meta-Llama-3.1-8B-Instruct/ initial_load_in_hf: true