tests/models/test_models_vq.py · 856dad57bb7a9ee13af4a08492e524b0a145a2c5 · 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_vq.py 3.03 KB

Replace test_models_vq.py