Skip to main content

1. Model Introduction

Wan2.1 is a broad open video family covering text-to-video and image-to-video across 1.3B and 14B checkpoints. Its practical strengths are motion-rich generation, temporal consistency, and readable Chinese/English text, with 480p and 720p variants for different quality and memory targets. Choose the 1.3B T2V model for consumer-GPU experiments and the 14B models when quality matters more than footprint. Wan2.1 is a dense DiT family; for timestep-specialized MoE capacity or the unified 5B TI2V path, use Wan2.2 instead.

2. SGLang-diffusion Installation

SGLang-diffusion offers multiple installation methods. You can choose the most suitable installation method based on your hardware platform and requirements. Please refer to the official SGLang-diffusion installation guide for installation instructions.

3. Model Deployment

This section provides deployment configurations optimized for different hardware platforms and use cases.

3.1 Basic Configuration

The Wan2.1 series offers models in multiple sizes and resolutions. SGLang supports Wan2.1 deployment on NVIDIA B200, B300, H200, H100, and AMD MI300X, MI325X, MI355X GPUs and Ascend A2/A3 Series NPUs. The recommended launch configurations vary by hardware, model size, and memory headroom. Interactive Command Generator: Use the configuration selector below to automatically generate an appropriate deployment command for your model variant and options.

3.2 Configuration Tips

Current supported optimization options are listed in the SGLang diffusion support matrix.
  • --vae-path: Path to a custom VAE model or HuggingFace model ID. If not specified, the VAE will be loaded from the main model path.
  • --num-gpus {NUM_GPUS}: Number of GPUs to use.
  • --tp-size {TP_SIZE}: Tensor parallelism size (for the encoder/DiT; keep (\leq 1) if relying heavily on CPU offload).
  • --sp-degree {SP_SIZE}: Sequence parallelism degree.
  • --ulysses-degree {ULYSSES_DEGREE}: Degree of DeepSpeed-Ulysses-style SP in USP.
  • --ring-degree {RING_DEGREE}: Degree of ring attention-style SP in USP.
  • --text-encoder-cpu-offload, --dit-cpu-offload, --vae-cpu-offload: Use CPU offload to reduce peak GPU memory when needed.

4. Model Invocation

4.1 Basic Usage

For more API usage and request examples, please refer to: SGLang Diffusion OpenAI API

4.1.1 Launch a server and then send requests

Command

4.1.2 Generate a video without launching a server

Command

4.2 Advanced Usage

4.2.1 Cache-DiT Acceleration

SGLang integrates Cache-DiT, a caching acceleration engine for Diffusion Transformers (DiT), to achieve significant inference speedups with minimal quality loss. You can set SGLANG_CACHE_DIT_ENABLED=True to enable it. For more details, please refer to the SGLang Cache-DiT documentation. Basic Usage
Command
Advanced Usage Combined Configuration Example:
Command

4.2.2 GPU Optimization

  • --dit-cpu-offload: Use CPU offload for DiT inference. Enable if you run out of memory with FSDP.
  • --text-encoder-cpu-offload: Use CPU offload for text encoder inference.
  • --image-encoder-cpu-offload: Use CPU offload for image encoder inference.
  • --vae-cpu-offload: Use CPU offload for VAE.
  • --pin-cpu-memory: Pin memory for CPU offload. Use as a workaround if you see “CUDA error: invalid argument”.

4.2.3 Supported LoRA Registry

SGLang supports applying Wan2.1 LoRA adapters on top of base models:
origin modelsupported LoRA
Wan-AI/Wan2.1-T2V-14BNIVEDAN/wan2.1-lora
Wan-AI/Wan2.1-I2V-14B-720Pvaliantcat/Wan2.1-Fight-LoRA
Example:
Command

5. Reference Benchmark

The following benchmark is a point-in-time reference for one model, hardware platform, SGLang image, and parameter set. It is not a complete hardware support matrix. Test Environment:
  • Hardware: AMD MI300X GPU (1x)
  • Model: Wan-AI/Wan2.1-T2V-14B-Diffusers
  • SGLang Docker Image Version: 0.5.9

5.1 How to Run Benchmarks with SGLang

You can use the built-in SGLang diffusion benchmark script to evaluate Wan2.1 performance on your hardware.

5.1.1 Generate a single video

Server Command:
Command
Benchmark Command:
Command
Result:
Output

5.1.2 Generate videos with Cache-DiT acceleration

Server Command:
Command
Benchmark Command:
Command
Result:
Output