From 1e86404456e872bfb937bd0820bc65239ad2e953 Mon Sep 17 00:00:00 2001 From: Didan Deng <33117903+wtomin@users.noreply.github.com> Date: Wed, 25 Feb 2026 15:43:08 +0800 Subject: [PATCH] [Doc] Refine Diffusion Tutorial Documents (#1305) Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com> --- docs/.nav.yml | 11 +++++------ docs/contributing/model/adding_diffusion_model.md | 12 ++++++------ .../features => design/feature}/cache_dit.md | 2 +- .../features => design/feature}/cfg_parallel.md | 2 +- .../features => design/feature}/sequence_parallel.md | 2 +- .../features => design/feature}/teacache.md | 2 +- .../features => design/feature}/tensor_parallel.md | 2 +- 7 files changed, 16 insertions(+), 17 deletions(-) rename docs/{contributing/features => design/feature}/cache_dit.md (99%) rename docs/{contributing/features => design/feature}/cfg_parallel.md (99%) rename docs/{contributing/features => design/feature}/sequence_parallel.md (99%) rename docs/{contributing/features => design/feature}/teacache.md (99%) rename docs/{contributing/features => design/feature}/tensor_parallel.md (99%) diff --git a/docs/.nav.yml b/docs/.nav.yml index 07db1b465..fa06b53ec 100644 --- a/docs/.nav.yml +++ b/docs/.nav.yml @@ -58,12 +58,6 @@ nav: - contributing/model/README.md - contributing/model/adding_omni_model.md - contributing/model/adding_diffusion_model.md - - Advanced Features: - - contributing/features/cfg_parallel.md - - contributing/features/sequence_parallel.md - - contributing/features/tensor_parallel.md - - contributing/features/cache_dit.md - - contributing/features/teacache.md - CI: contributing/ci - Design Documents: - design/index.md @@ -72,6 +66,11 @@ nav: - design/feature/disaggregated_inference.md - design/feature/ray_based_execution.md - design/feature/omni_connectors/ + - design/feature/cfg_parallel.md + - design/feature/sequence_parallel.md + - design/feature/tensor_parallel.md + - design/feature/cache_dit.md + - design/feature/teacache.md - Module Design: - design/module/ar_module.md - design/module/dit_module.md diff --git a/docs/contributing/model/adding_diffusion_model.md b/docs/contributing/model/adding_diffusion_model.md index 10755c6c6..c1094802c 100644 --- a/docs/contributing/model/adding_diffusion_model.md +++ b/docs/contributing/model/adding_diffusion_model.md @@ -1,4 +1,4 @@ -# Adding a Diffusion Model to vLLM-Omni +# Adding a Diffusion Model This guide walks you through adding a new diffusion model to vLLM-Omni. We use **Qwen-Image** as the primary example, with references to other models (LongCat, Flux, Wan2.2) to illustrate different patterns. @@ -680,7 +680,7 @@ vLLM-Omni automatically compiles blocks in `_repeated_blocks` when `torch.compil ### Tensor Parallelism -See detailed guide: [How to add Tensor Parallel support](../features/tensor_parallel.md) +See detailed guide: [How to add Tensor Parallel support](../../design/feature/tensor_parallel.md) **Quick setup:** @@ -694,7 +694,7 @@ omni = Omni(model="your-model", tensor_parallel_size=2) ### CFG Parallelism -See detailed guide: [How to add CFG-Parallel support](../features/cfg_parallel.md) +See detailed guide: [How to add CFG-Parallel support](../../design/feature/cfg_parallel.md) **Quick setup:** @@ -708,7 +708,7 @@ omni = Omni(model="your-model", cfg_parallel_size=2) ### Sequence Parallelism -See detailed guide: [How to add Sequence Parallel support](../features/sequence_parallel.md) +See detailed guide: [How to add Sequence Parallel support](../../design/feature/sequence_parallel.md) **Quick setup:** @@ -724,7 +724,7 @@ omni = Omni(model="your-model", ulysses_degree=2, ring_degree=2) #### TeaCache -See detailed guide: [How to add TeaCache support](../features/teacache.md) +See detailed guide: [How to add TeaCache support](../../design/feature/teacache.md) **Quick setup:** @@ -744,7 +744,7 @@ omni = Omni(model="your-model", #### Cache-DiT -See detailed guide: [How to add Cache-DiT support](../features/cache_dit.md) +See detailed guide: [How to add Cache-DiT support](../../design/feature/cache_dit.md) **Quick setup:** diff --git a/docs/contributing/features/cache_dit.md b/docs/design/feature/cache_dit.md similarity index 99% rename from docs/contributing/features/cache_dit.md rename to docs/design/feature/cache_dit.md index f39b93578..3112bb591 100644 --- a/docs/contributing/features/cache_dit.md +++ b/docs/design/feature/cache_dit.md @@ -1,4 +1,4 @@ -# Support Cache-DiT +# Cache-DiT This section describes how to add cache-dit acceleration to a new diffusion pipeline. We use the Qwen-Image pipeline and LongCat-Image pipeline as reference implementations. diff --git a/docs/contributing/features/cfg_parallel.md b/docs/design/feature/cfg_parallel.md similarity index 99% rename from docs/contributing/features/cfg_parallel.md rename to docs/design/feature/cfg_parallel.md index b169f13d0..e31e64edd 100644 --- a/docs/contributing/features/cfg_parallel.md +++ b/docs/design/feature/cfg_parallel.md @@ -1,4 +1,4 @@ -# Support CFG-Parallel +# CFG-Parallel This section describes how to add CFG-Parallel (Classifier-Free Guidance Parallel) to a diffusion pipeline. We use the Qwen-Image pipeline as the reference implementation. diff --git a/docs/contributing/features/sequence_parallel.md b/docs/design/feature/sequence_parallel.md similarity index 99% rename from docs/contributing/features/sequence_parallel.md rename to docs/design/feature/sequence_parallel.md index f90ac817f..c5c667885 100644 --- a/docs/contributing/features/sequence_parallel.md +++ b/docs/design/feature/sequence_parallel.md @@ -1,4 +1,4 @@ -# Support Sequence Parallel +# Sequence Parallel This section describes how to add Sequence Parallel (SP) to a diffusion transformer model. We use the Qwen-Image transformer and Wan2.2 transformer as reference implementations. diff --git a/docs/contributing/features/teacache.md b/docs/design/feature/teacache.md similarity index 99% rename from docs/contributing/features/teacache.md rename to docs/design/feature/teacache.md index 6c29c0f99..775fb635b 100644 --- a/docs/contributing/features/teacache.md +++ b/docs/design/feature/teacache.md @@ -1,4 +1,4 @@ -# Support TeaCache +# TeaCache This section describes how to add TeaCache to a diffusion transformer model. We use the Qwen-Image transformer as the reference implementation. diff --git a/docs/contributing/features/tensor_parallel.md b/docs/design/feature/tensor_parallel.md similarity index 99% rename from docs/contributing/features/tensor_parallel.md rename to docs/design/feature/tensor_parallel.md index 9d25a6e2d..bcafde7e7 100644 --- a/docs/contributing/features/tensor_parallel.md +++ b/docs/design/feature/tensor_parallel.md @@ -1,4 +1,4 @@ -# Support Tensor Parallel +# Tensor Parallel This section describes how to add Tensor Parallel (TP) to a diffusion transformer model. We use the Z-Image transformer as the reference implementation.