From 10919b32cafe51a78caf3faaacb9f6d8fdc0c725 Mon Sep 17 00:00:00 2001 From: taoliu49 Date: Fri, 21 Aug 2026 15:51:47 +0800 Subject: [PATCH] fix: preserve incomplete flash checkpoints --- swift/trainers/mixin.py | 1 - 1 file changed, 1 deletion(-) diff --git a/swift/trainers/mixin.py b/swift/trainers/mixin.py index 578e4a41b8..6ab847ff68 100644 --- a/swift/trainers/mixin.py +++ b/swift/trainers/mixin.py @@ -699,7 +699,6 @@ def _save_flash_checkpoint(self, model, trial, metrics=None): if not success: logger.info(f'Skip saving the checkpoint of step {self.state.global_step} ' 'because the latest checkpoint is not finished.') - shutil.rmtree(output_dir, ignore_errors=True) if self.args.push_to_hub: self._push_from_checkpoint(output_dir)