fix path_or_state bug

dalle2_pytorch/dalle2_pytorch.py

@@ -2589,7 +2589,7 @@ class Decoder(nn.Module):
                 if is_inpaint and not (is_last_timestep or is_last_resample_step):
                     # in repaint, you renoise and resample up to 10 times every step
                     time_next_cond = torch.full((batch,), time_next, device = device, dtype = torch.long)
-                    img = noise_scheduler.q_sample_from_to(img, time_cond, time_next_cond)
+                    img = noise_scheduler.q_sample_from_to(img, time_next_cond, time_cond)
 
         if exists(inpaint_image):
             img = (img * ~inpaint_mask) + (inpaint_image * inpaint_mask)

dalle2_pytorch/trainer.py

@@ -300,7 +300,7 @@ class DiffusionPriorTrainer(nn.Module):
 
         # all processes need to load checkpoint. no restriction here
         if isinstance(path_or_state, str):
-            path = Path(path)
+            path = Path(path_or_state)
             assert path.exists()
             loaded_obj = torch.load(str(path), map_location=self.device)
 

dalle2_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.0.4'
+__version__ = '1.0.6'