Edna 15ca813e3e Add transformer tests

2025-06-12 03:45:43 -06:00

..

__init__.py

[Core] move transformer scripts to transformers modules (#6747)

2024-01-29 22:28:28 +05:30

test_models_dit_transformer2d.py

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

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

test_models_pixart_transformer2d.py

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

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

test_models_prior.py

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

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

test_models_transformer_allegro.py

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

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

test_models_transformer_aura_flow.py

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

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

test_models_transformer_chroma.py

Add transformer tests

2025-06-12 03:45:43 -06:00

test_models_transformer_cogvideox.py

Fix offload tests for CogVideoX and CogView3 (#10547)

2025-01-15 12:24:46 +05:30

test_models_transformer_cogview3plus.py

Fix offload tests for CogVideoX and CogView3 (#10547)

2025-01-15 12:24:46 +05:30

test_models_transformer_cogview4.py

CogView4 (supports different length c and uc) (#10649)

2025-02-15 21:46:48 +05:30

test_models_transformer_consisid.py

[core] ConsisID (#10140)

2025-01-19 13:10:08 +05:30

test_models_transformer_cosmos.py

Cosmos (#10660)

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

test_models_transformer_easyanimate.py

Add EasyAnimateV5.1 text-to-video, image-to-video, control-to-video generation model (#10626)

2025-03-03 18:37:19 +05:30

test_models_transformer_flux.py

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

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

test_models_transformer_hidream.py

[tests] Enable testing for HiDream transformer (#11478)

2025-05-13 21:01:20 +05:30

test_models_transformer_hunyuan_dit.py

[Tests] Improve transformers model test suite coverage - Hunyuan DiT (#8916)

2024-08-06 12:59:30 +05:30

test_models_transformer_hunyuan_video_framepack.py

[tests] add tests for framepack transformer model. (#11520)

2025-05-11 09:50:06 +05:30

test_models_transformer_hunyuan_video.py

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

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

test_models_transformer_latte.py

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

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

test_models_transformer_ltx.py

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

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

test_models_transformer_lumina2.py

follow-up refactor on lumina2 (#10776)

2025-02-14 14:57:27 -10:00

test_models_transformer_lumina.py

[CI] More Fast GPU Test Fixes (#9346)

2024-09-03 13:22:38 +05:30

test_models_transformer_mochi.py

[Tests] Fix more tests sayak (#10359)

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

test_models_transformer_omnigen.py

[CI] Fix Fast GPU tests on PR (#10912)

2025-02-27 14:21:50 +05:30

test_models_transformer_sana.py

fix offload gpu tests etc (#10366)

2025-01-21 18:52:36 +05:30

test_models_transformer_sd3.py

Update Ruff to latest Version (#10919)

2025-04-09 16:51:34 +05:30

test_models_transformer_temporal.py

[Tests] Improve transformers model test suite coverage - Temporal Transformer (#8932)

2024-07-23 15:36:30 +05:30

test_models_transformer_wan.py

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

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