Sayak Paul c80eda9d3e [Tests] Test layerwise casting with training (#10765) ...

* add a test to check if we can train with layerwise casting.

* updates

* updates

* style

2025-02-11 16:02:28 +05:30

..

__init__.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_unet_1d.py

[Tests] Test layerwise casting with training (#10765)

2025-02-11 16:02:28 +05:30

test_models_unet_2d_condition.py

[chore] remove prints from tests. (#10505)

2025-01-09 14:19:43 +05:30

test_models_unet_2d_flax.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_unet_2d.py

[core] Layerwise Upcasting (#10347)

2025-01-22 19:49:37 +05:30

test_models_unet_3d_condition.py

[Tests] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30

test_models_unet_controlnetxs.py

[Tests] skip tests properly with unittest.skip() (#10527)

2025-01-11 08:46:22 +05:30

test_models_unet_motion.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_unet_spatiotemporal.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_unet_2d_blocks.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_unet_blocks_common.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00