Sayak Paul d70f8ee18b [WAN] fix recompilation issues (#11475) ...

* [tests] Add torch.compile() test for WanTransformer3DModel

* fix wan recompilation issues.

* style

---------

Co-authored-by: tongyu0924 <winnie920924@gmail.com>

2025-04-30 20:29:08 -10:00

..

autoencoders

make autoencoders. controlnet_flux and wan_transformer3d_single_file pass on xpu (#11461)

2025-05-01 02:43:31 +05:30

transformers

[WAN] fix recompilation issues (#11475)

2025-04-30 20:29:08 -10:00

unets

Update Ruff to latest Version (#10919)

2025-04-09 16:51:34 +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] fix import. (#11434)

2025-04-28 13:32:28 +08:00