src/diffusers/models/unet_2d_blocks.py · 4125756e88e82370c197fecf28e9f0b4d7eee6c3 · renzhc / diffusers_dcu

Refactor cross attention and allow mechanism to tweak cross attention function (#1639) · 4125756e

Patrick von Platen authored Dec 20, 2022



* first proposal

* rename

* up

* Apply suggestions from code review

* better

* up

* finish

* up

* rename

* correct versatile

* up

* up

* up

* up

* fix

* Apply suggestions from code review

* make style

* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* add error message
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

4125756e

unet_2d_blocks.py 77.5 KB

Replace unet_2d_blocks.py