DDIM resolution->image_size

src/diffusers/pipelines/ddim/pipeline_ddim.py

@@ -36,7 +36,7 @@ class DDIMPipeline(DiffusionPipeline):
 
         # Sample gaussian noise to begin loop
         image = torch.randn(
-            (batch_size, self.unet.in_channels, self.unet.resolution, self.unet.resolution),
+            (batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size),
             generator=generator,
         )
         image = image.to(torch_device)