Sayak Paul 4af76d0d7d [tests] Changes to the torch.compile() CI and tests (#11508) ...

* remove compile cuda docker.

* replace compile cuda docker path.

* better manage compilation cache.

* propagate similar to the pipeline tests.

* remove unneeded compile test.

* small.

* don't check for deleted files.

2025-05-26 08:31:04 -07:00

..

autoencoders

Cosmos (#10660)

2025-05-07 20:59:09 +05:30

transformers

[tests] Changes to the torch.compile() CI and tests (#11508)

2025-05-26 08:31:04 -07:00

unets

[tests] Add torch.compile test for UNet2DConditionModel (#11537)

2025-05-14 11:26:12 +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] Changes to the torch.compile() CI and tests (#11508)

2025-05-26 08:31:04 -07:00

test_models_auto.py

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

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