0.1.5

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

@@ -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 = Path(results_folder)
-            self.results_folder.mkdir(exist_ok = True)
+        self.results_folder.mkdir(exist_ok = True)
 
         # step counter state
 

denoising_diffusion_pytorch/version.py

@@ -1 +1 @@
-__version__ = '0.1.0'
+__version__ = '0.1.5'