Patrick von Platen a80480f0f2 [Tests] Improve unet / vae tests (#1018) ... * improve tests * up * finish * upload * add init * up * finish vae * finish * reduce loading time with device_map * remove device_map from CPU * uP		2022-10-28 13:43:26 +02:00
..
__init__.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_unet_1d.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_unet_2d.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_vae.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_vae_flax.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_vq.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00