Skip to content

[Feature] Support VAE tiling parallel encode for LTX-2 #4831

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
abtonmoy wants to merge 11 commits into
base: main
Choose a base branch
from
+992 −20
Open

[Feature] Support VAE tiling parallel encode for LTX-2 #4831

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
f2e653d
[Feature] Support VAE tiling parallel encode for LTX-2
abtonmoy Jul 2, 2026
b20ebf1
[Doc] Document the encode-tile contract on DistributedAutoencoderKLLT…
abtonmoy Jul 4, 2026
089dbb2
[CI] Cover the non-divisible tile fallback in encode merge equivalence
abtonmoy Aug 1, 2026
95f7db2
[CI] Add LTX-2.5 I2V VAE patch-parallel e2e case
abtonmoy Sep 8, 2026
21cf0c9
[Test] Add CPU/Gloo integration tests for the LTX2 VAE encode executor
abtonmoy Sep 13, 2026
5518f32
Merge branch 'main' into feat/ltx2-vae-encode-parallel
abtonmoy Oct 4, 2026
86bf16e
ci: run test_ltx25.py in the weekly LTX 2-GPU step
abtonmoy Oct 16, 2026
7d99b51
Merge remote-tracking branch 'origin/main' into feat/ltx2-vae-encode-…
abtonmoy Oct 16, 2026
1d22a9b
[Misc] Satisfy main's mypy and SPDX pre-commit hooks after merging main
abtonmoy Oct 16, 2026
548f86f
[CI/Build] Select the CPU platform in the LTX-2 encode gloo workers
abtonmoy Oct 16, 2026
17d5c8e
[Bugfix] Agree on rank-local failures in the LTX-2 known-metadata VAE…
abtonmoy Oct 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .buildkite/cuda/test-weekly.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -598,6 +598,7 @@ steps:
- >-
pytest -sv
tests/e2e/online_serving/test_ltx.py
tests/e2e/online_serving/test_ltx25.py
-m "slow and diffusion and H100 and cards_2"
--run-level "full_model"
mirror_hardwares: h100_2
Expand Down
1 change: 1 addition & 0 deletions docs/design/feature/vae_parallel.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -466,6 +466,7 @@ Complete examples in the codebase:
| **Wan2.2** | `vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_wan.py` | ✅ | ✅ |
| **Qwen-Image** | `vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_qwenimage.py` | ✅ | ❌ |
| **FLUX.2-dev** | `vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_flux2.py` | ✅ | ✅ |
| **LTX-2 / LTX-2.3** | `vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_ltx2.py` | ✅ | ✅ |

---

Expand Down
6 changes: 3 additions & 3 deletions docs/user_guide/diffusion_features.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -194,9 +194,9 @@ The following tables show which models support each feature:
| **ABot-World** | ❌ | ❌ | ❌ | ❌ | ✅ | ❌ | ❌ | ❓ | ❌ | ❌ | ✅ (one request) |
| **Wan2.2-S2V** | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (encode/decode) | ❌ | ❌ |
| **Wan2.1-VACE** | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (decode) | ❌ | ❌ |
| **LTX-2** | ❌ | ✅ | ✅ (Ulysses only) | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (decode) | ❌ | ❌ |
| **LTX-2.3** | ❌ | ✅ | ✅ (Ulysses only) | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (decode) | ❌ | ❌ |
| **LTX-2.5** | ❌ | ❓ (one-stage) | ✅* (Ulysses only) | ✅ (Full only) | ❓ | ❌ | ✅ | ✅ | ✅ (decode) | ❓ (FP8) | ❌ |
| **LTX-2** | ❌ | ✅ | ✅ (Ulysses only) | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (encode/decode) | ❌ | ❌ |
| **LTX-2.3** | ❌ | ✅ | ✅ (Ulysses only) | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (encode/decode) | ❌ | ❌ |
| **LTX-2.5** | ❌ | ❓ (one-stage) | ✅* (Ulysses only) | ✅ (Full only) | ❓ | ❌ | ✅ | ✅ | ✅ (encode/decode) | ❓ (FP8) | ❌ |
| **SANA-Video-2B T2V I2V** | ❌ | ❌ | ✅ (`--usp`, frame-sharded) | ✅ | ✅ (TP=2 only) | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ |
| **Helios** | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ❌ | ❌ | ✅* |
| **HunyuanVideo-1.5 T2V I2V** | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ✅ | ✅ | ✅ (encode/decode) | ✅ | ❌ |
Expand Down
Loading
Loading