Remove duplicate save when triggering save next step
This commit is contained in:
parent
7ee1f98f6d
commit
1e12b6b73f
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue