Merge pull request #1162 from bghira/bugfix/rmtree-errors

chore: ignore rmtree errors
bghira · Nov 16, 2024 · ea620b2 · ea620b2
2 parents 13799b6 + 12f1c16
commit ea620b2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/helpers/training/save_hooks.py b/helpers/training/save_hooks.py
@@ -325,7 +325,7 @@ def _save_full_model(self, models, weights, output_dir):
                 shutil.copy2(s, d)
 
         # Remove the temporary directory
-        shutil.rmtree(temporary_dir)
+        shutil.rmtree(temporary_dir, ignore_errors=True)
 
     def save_model_hook(self, models, weights, output_dir):
         # Write "training_state.json" to the output directory containing the training state

diff --git a/helpers/training/trainer.py b/helpers/training/trainer.py
@@ -2717,7 +2717,7 @@ def train(self):
                                             self.config.output_dir, removing_checkpoint
                                         )
                                         try:
-                                            shutil.rmtree(removing_checkpoint)
+                                            shutil.rmtree(removing_checkpoint, ignore_errors=True)
                                         except Exception as e:
                                             logger.error(
                                                 f"Failed to remove directory: {removing_checkpoint}"