tests/models/test_models_unet_2d.py · 8d6487f3cbe89bb6e32f82fc9f04df6ce001ef24 · renzhc / diffusers_dcu

[Tests] Improve unet / vae tests (#1018) · a80480f0

Patrick von Platen authored Oct 28, 2022

* improve tests

* up

* finish

* upload

* add init

* up

* finish vae

* finish

* reduce loading time with device_map

* remove device_map from CPU

* uP

a80480f0

test_models_unet_2d.py 21.8 KB

Replace test_models_unet_2d.py