tests/models/test_models_unet_2d.py · ac3fc649066df9d347df519b3d0877d41fb847b1 · 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

test_models_unet_2d.py 31.5 KB

Replace test_models_unet_2d.py