fix repaint

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_next_cond, time_cond)
+                    img = noise_scheduler.q_sample_from_to(img, time_cond, time_next_cond)
 
         if exists(inpaint_image):
             img = (img * ~inpaint_mask) + (inpaint_image * inpaint_mask)

dalle2_pytorch/version.py

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