How can I deal with images that height is not equal to weight?

[Mystere271]

Hi Lucidrains,

I noticed that height and width of image must be {img_size}, but what if my image's weight and height are different? Does it mean that I have to clip the images? Or is there any more elegant way to handle it?

Thanks!

[Mut1nyJD]

No simply change the parameters to a tuple and change the asserts to check on a tuple. Rest will continue working, the UNet does not care if it is square or has a different aspect ratio. At least it worked for me

[zuiwomeirenxi]

No simply change the parameters to a tuple and change the asserts to check on a tuple. Rest will continue working, the UNet does not care if it is square or has a different aspect ratio. At least it worked for me

hi cloud you please tell me how about set output size of generative picture?