Compare commits

..
1 Commits
2 changed files with 3 additions and 3 deletions
@@ -810,8 +810,8 @@ class Trainer(object):
if self.accelerator.is_main_process:
self.ema = EMA(diffusion_model, beta = ema_decay, update_every = ema_update_every)
self.results_folder = Path(results_folder)
self.results_folder.mkdir(exist_ok = True)
self.results_folder = Path(results_folder)
self.results_folder.mkdir(exist_ok = True)
# step counter state
+1 -1
View File
@@ -1 +1 @@
__version__ = '0.1.5'
__version__ = '1.16.6'