Revert "checkpoint model on first step callback (#2906)"

This reverts commit 10ba1622f7.
2025-07-15 15:01:12 -04:00
parent 10ba1622f7
commit 6f6d917a99
146 changed files with 9 additions and 419 deletions
--- a/examples/mistral/mistral-dpo-qlora.yml
+++ b/examples/mistral/mistral-dpo-qlora.yml
@@ -80,5 +80,3 @@ weight_decay: 0.0
 special_tokens:
  bos_token: "<|im_start|>"
  eos_token: "<|im_end|>"
-
-# save_first_step: true  # uncomment this to validate checkpoint saving works with your config