Sayak Paul 91545666e0 [tests] model-level device_map clarifications (#11681) ...

* add clarity in documentation for device_map

* docs

* fix how compiler tester mixins are used.

* propagate

* more

* typo.

* fix tests

* fix order of decroators.

* clarify more.

* more test cases.

* fix doc

* fix device_map docstring in pipeline_utils.

* more examples

* more

* update

* remove code for stuff that is already supported.

* fix stuff.

2025-06-11 22:41:59 +05:30

..

autoencoders

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

2025-06-11 15:17:06 +05:30

transformers

[tests] Fix how compiler mixin classes are used (#11680)

2025-06-09 09:24:45 +05:30

unets

[tests] model-level device_map clarifications (#11681)

2025-06-11 22:41:59 +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

[Tests] Fix more tests sayak (#10359)

2024-12-23 19:11:21 +05:30

test_layers_utils.py

Support bfloat16 for Upsample2D (#9480)

2024-10-01 16:08:12 -10:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

[tests] add test for torch.compile + group offloading (#11670)

2025-06-06 11:34:44 +05:30

test_models_auto.py

[Feature] AutoModel can load components using model_index.json (#11401)

2025-05-26 14:06:36 +05:30