src/diffusers/models/unet_2d_condition.py · 5c7a35a25915f29aa79e5b69d831fd0f7d7d8d41 · renzhc / diffusers_dcu

[Torch 2.0 compile] Fix more torch compile breaks (#3313) · 5c7a35a2

Patrick von Platen authored May 02, 2023



* Fix more torch compile breaks

* add tests

* Fix all

* fix controlnet

* fix more

* Add Horace He as co-author.
>
>
Co-authored-by: Horace He <horacehe2007@yahoo.com>

* Add Horace He as co-author.
Co-authored-by: Horace He <horacehe2007@yahoo.com>

---------
Co-authored-by: Horace He <horacehe2007@yahoo.com>

5c7a35a2

unet_2d_condition.py 37.9 KB

Replace unet_2d_condition.py