diff --git a/examples/llama-3/qlora-1b-kto.yaml b/examples/llama-3/qlora-1b-kto.yaml index fbcf80e9b..32168dc37 100644 --- a/examples/llama-3/qlora-1b-kto.yaml +++ b/examples/llama-3/qlora-1b-kto.yaml @@ -55,7 +55,7 @@ tf32: true gradient_checkpointing: true gradient_checkpointing_kwargs: - use_reentrant: fal + use_reentrant: false early_stopping_patience: resume_from_checkpoint: local_rank: