Suraj Patil f07a16e09b update unet2d (#1376) ...

* boom boom

* remove duplicate arg

* add use_linear_proj arg

* fix copies

* style

* add fast tests

* use_linear_proj -> use_linear_projection

2022-11-23 20:46:30 +01:00

..

__init__.py

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

2022-10-28 13:43:26 +02:00

test_models_unet_1d.py

Add Scheduler.from_pretrained and better scheduler changing (#1286)

2022-11-15 18:15:13 +01:00

test_models_unet_2d.py

update unet2d (#1376)

2022-11-23 20:46:30 +01:00

test_models_vae_flax.py

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

2022-10-28 13:43:26 +02:00

test_models_vae.py

default fast model loading 🔥 (#1115)

2022-11-03 17:25:57 +01:00

test_models_vq.py

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

2022-10-28 13:43:26 +02:00