Question about the image size

[DongyangHuLi]

Thanks for sharing the code! I'm confused about the size of the image input from the diffusion model. In the code why limit the height and width of the image must be equal, can not equal? Looking forward to your reply!

[w86763777]

The height and width of the image do not need to be equal, and this can be accomplished by updating the UNet model.

Another thing to consider is the downsampling and upsampling of UNet, which may cause the input size to differ from the output size if the dimensions are not divisible by 2.