diff --git a/extensions_built_in/sd_trainer/DiffusionTrainer.py b/extensions_built_in/sd_trainer/DiffusionTrainer.py index 51c081b3..59a264c6 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: