Prathik Rao de1426119d Make UNet2DConditionOutput pickle-able (#3857) ...

* add default to unet output to prevent it from being a required arg

* add unit test

* make style

* adjust unit test

* mark as fast test

* adjust assert statement in test

---------

Co-authored-by: Prathik Rao <prathikrao@microsoft.com@orttrainingdev8.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Co-authored-by: root <root@orttrainingdev8.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>

2023-07-06 13:12:41 +05:30

..

__init__.py

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

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

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

relax tol attention conversion test (#3842)

2023-06-21 12:35:38 -07:00

test_layers_utils.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_lora_layers.py

Add draft for lora text encoder scale (#3626)

2023-06-06 22:47:46 +01:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_modeling_common.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_models_prior.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_models_unet_1d.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_models_unet_2d_condition.py

Make UNet2DConditionOutput pickle-able (#3857)

2023-07-06 13:12:41 +05:30

test_models_unet_2d_flax.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_models_unet_2d.py

Correct bad attn naming (#3797)

2023-06-22 13:52:48 +02:00

test_models_unet_3d_condition.py

Improve memory text to video (#3930)

2023-07-03 18:17:34 +02:00

test_models_vae_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_models_vae.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_models_vq.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_unet_2d_blocks.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_unet_blocks_common.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00