dg845 caae16768a Move Video and Audio Text Encoder Connectors to Transformer (#12) ...

* Denormalize audio latents in I2V pipeline (analogous to T2V change)

* Initial refactor to put video and audio text encoder connectors in transformer

* Get LTX 2 transformer tests working after connector refactor

* precompute run_connectors,.

* fixes

* Address review comments

* Calculate RoPE double precisions freqs using torch instead of np

* Further simplify LTX 2 RoPE freq calc

* Make connectors a separate module (#18)

* remove text_encoder.py

* address yiyi's comments.

* up

* up

* up

* up

---------

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

2026-01-05 20:11:13 +05:30

..

__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

Move Video and Audio Text Encoder Connectors to Transformer (#12)

2026-01-05 20:11:13 +05:30

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

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

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

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