src/diffusers/models/unet_2d_condition.py · ab0e92fdc8b42e1739840d10bb0b1dc7edaf2ff3 · 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_condition.py 22 KB

Replace unet_2d_condition.py