Sayak Paul 05e867784d [tests] device_map tests for all models. (#11708) ...

* device_map tests for all models.

* updates

* Update tests/models/test_modeling_common.py

Co-authored-by: Aryan <aryan@huggingface.co>

* fix device_map in test

---------

Co-authored-by: Aryan <aryan@huggingface.co>

2025-06-18 10:52:06 +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

[tests] device_map tests for all models. (#11708)

2025-06-18 10:52:06 +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

enable torchao test cases on XPU and switch to device agnostic APIs for test cases (#11654)

2025-06-11 15:17:06 +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