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-qlora-fsdp.yml
+++ b/examples/mistral/mistral-qlora-fsdp.yml
@@ -74,5 +74,3 @@ fsdp_config:
  fsdp_state_dict_type: FULL_STATE_DICT
  fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
 special_tokens:
-
-# save_first_step: true  # uncomment this to validate checkpoint saving works with your config