fix

Author: ankitageorge

torchtitan/components/checkpoint.py

@@ -699,8 +699,10 @@ def _save_last_step(self, curr_step: int) -> None:
         else:
             logger.info(f"Saving a full checkpoint at last step, step {curr_step}.")
 
-        self.save_with_gc(
-            self.states, checkpoint_id=self._create_checkpoint_id(curr_step)
+        save_with_gc(
+            self.states,
+            checkpoint_id=self._create_checkpoint_id(curr_step),
+            hf_safetensors_format=self.enable_hf_safetensors_format,
         )
 
     def _should_save(self, curr_step: int, last_step: bool = False) -> bool: