tests/models/test_models_unet_1d.py · ef2ea33c3bc061fffa8bc4ccd640306ca1a1847d · 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_1d.py 1.43 KB

Replace test_models_unet_1d.py