Kashif Rasul d54669a73e [Qwen] avoid creating attention masks when there is no padding (#12987) ...

* avoid creating attention masks when there is no padding

* make fix-copies

* torch compile tests

* set all ones mask to none

* fix positional encoding from becoming > 4096

* fix from review

* slice freqs_cis to match the input sequence length

* keep only attenton masking change

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2026-01-27 12:42:48 -10:00

..

__init__.py

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

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

test_models_dit_transformer2d.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_pixart_transformer2d.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_prior.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_allegro.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_aura_flow.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_bria_fibo.py

Bria fibo (#12545)

2025-10-28 16:27:48 +05:30

test_models_transformer_bria.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_chroma.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_cogvideox.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_cogview3plus.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_cogview4.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_consisid.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_cosmos.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_easyanimate.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_flux2.py

let's go Flux2 🚀 (#12711)

2025-11-25 21:49:04 +05:30

test_models_transformer_flux.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_hidream.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_hunyuan_1_5.py

[tests] fix hunuyanvideo 1.5 offloading tests. (#12782)

2025-12-03 18:07:59 +05:30

test_models_transformer_hunyuan_dit.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_hunyuan_video_framepack.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_hunyuan_video.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_latte.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_ltx2.py

Add LTX 2.0 Video Pipelines (#12915)

2026-01-07 21:24:27 -08:00

test_models_transformer_ltx.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_lumina2.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_lumina.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_mochi.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_omnigen.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_prx.py

Prx (#12525)

2025-10-21 17:09:22 -07:00

test_models_transformer_qwenimage.py

[Qwen] avoid creating attention masks when there is no padding (#12987)

2026-01-27 12:42:48 -10:00

test_models_transformer_sana_video.py

[SANA-Video] Adding 5s pre-trained 480p SANA-Video inference (#12584)

2025-11-05 21:08:47 -08:00

test_models_transformer_sana.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_sd3.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_skyreels_v2.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_temporal.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_wan_animate.py

[WIP]Add Wan2.2 Animate Pipeline (Continuation of #12442 by tolgacangoz) (#12526)

2025-11-12 16:52:31 -10:00

test_models_transformer_wan.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_models_transformer_z_image.py

[Z-Image] various small changes, Z-Image transformer tests, etc. (#12741)

2025-12-03 19:35:46 +05:30