Just a tiny change will allow sampling at different resolutions.

[hizzlehoff]

I am doing this locally but it would be great if the package had this small change to allow sampling at different resolutions:

Line 627 in denoising_diffusion_pytorch.py:
@torch.no_grad()
def sample(self, batch_size = 16, widthMultiplier = 1, heightMultiplier = 1):
image_size, channels = self.image_size, self.channels
sample_fn = self.p_sample_loop if not self.is_ddim_sampling else self.ddim_sample
return sample_fn((batch_size, channels, int(image_size * heightMultiplier), int(image_size * widthMultiplier) ))