src/diffusers/models/unet_2d_condition.py · 8fd3a74322befbd13bb461e4cb9e1a57f6e9ed96 · renzhc / diffusers_dcu

Fix using non-square images with UNet2DModel and DDIM/DDPM pipelines (#1289) · 8fd3a743

Penn authored Nov 23, 2022



* fix non square images with UNet2DModel and DDIM/DDPM pipelines

* fix unet_2d `sample_size` docstring

* update pipeline tests for unet uncond
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

8fd3a743

unet_2d_condition.py 15.3 KB

Replace unet_2d_condition.py