diff --git a/extensions_built_in/sd_trainer/DiffusionTrainer.py b/extensions_built_in/sd_trainer/DiffusionTrainer.py
index 51c081b..59a264c 100644
--- a/extensions_built_in/sd_trainer/DiffusionTrainer.py
+++ b/extensions_built_in/sd_trainer/DiffusionTrainer.py
@@ -202,7 +202,6 @@ class DiffusionTrainer(SDTrainer):
             flush()
             if self.progress_bar is not None:
                 self.progress_bar.unpause()
-            self.save(self.step_num)
 
     async def _update_key(self, key, value):
         if not self.accelerator.is_main_process: