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) ))