tests/models/test_models_unet_2d.py · 189e9f01b3f2d2fdb73455cdbae71ffd5cac0ee3 · chenpangpang / diffusers

"TensorFlow2x/vscode:/vscode.git/clone" did not exist on "aa90a6bfa0d47f525e722f9464f1900210c3dc37"

Test Cleanup Precision issues (#4812) · 189e9f01

Dhruv Nair authored Sep 01, 2023



* proposal for flaky tests

* more precision fixes

* move more tests to use cosine distance

* more test fixes

* clean up

* use default attn

* clean up

* update expected value

* make style

* make style

* Apply suggestions from code review

* Update src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py

* make style

* fix failing tests

---------
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

189e9f01

test_models_unet_2d.py 9.72 KB

Replace test_models_unet_2d.py