README.md

---
license: apache-2.0
base_model:
- Wan-AI/Wan2.2-T2V-A14B
- Wan-AI/Wan2.2-I2V-A14B
library_name: diffusers
tags:
- video_generation
- NVFP4
- Sparse_Attention
- Wan
---
# 🎬 LightWan2.2-A14B

> **An extremely efficient Wan 2.2 14B variant: NVFP4 Quantization-Aware Step Distillation with Sparse Attention for Blackwell Architecture**

[![GitHub](https://img.shields.io/badge/GitHub-ModelTC/LightX2V-blue)](https://github.com/ModelTC/LightX2V)
[![HuggingFace](https://img.shields.io/badge/HuggingFace-LightWan2.2--A14B-yellow)](https://huggingface.co/lightx2v/LightWan2.2-A14B)
[![Blog](https://img.shields.io/badge/Blog-LightWan2.2--A14B-green)](https://light-ai.top/LightX2V-BLOG/posts/LightWan22-A14B/)

## 📋 Table of Contents

- [✨ Features](#-features)
- [🚀 Quick Start](#-quick-start)
- [🎬 Generation Results](#-generation-results)
- [⚡ Performance Comparison](#-performance-comparison)
- [⚠️ Notes](#️-notes)
- [🤝 Community](#-community)

## ✨ Features

- **⚡ 4-Step Inference**: Two high-noise expert steps followed by two low-noise expert steps, enabling extremely fast Wan2.2 MoE generation on a single Blackwell GPU.
- **🎯 NVFP4 Quantization**: Quantization-aware step distillation reduces memory traffic and compute cost while targeting Blackwell architecture.
- **🧩 Sparse Attention**: Accelerates the costly O(n²) self-attention workload with sparse attention, reducing end-to-end latency for high-resolution video generation.
- **🔧 LightX2V Integration**: Recommended runtime stack for stable deployment and best performance.
- **🚀 High-Quality Generation**: Preserves the visual quality of Wan2.2-T2V/I2V-14B while dramatically improving inference speed.

## 🚀 Quick Start

We strongly recommend using the official LightX2V Docker image for the cleanest environment and best reproducibility.

### Option A: Docker Recommended

```bash
# 1. Pull LightX2V Docker image
docker pull lightx2v/lightx2v:26052801-cu130-5090

# 2. Run single-GPU inference
# Text-to-video
bash scripts/wan22/extreme/run_wan22_moe_t2v_extreme.sh
# Image-to-video
bash scripts/wan22/extreme/run_wan22_moe_i2v_extreme.sh

# 3. Run multi-GPU sequence-parallel inference
# Text-to-video
bash scripts/wan22/extreme/run_wan22_moe_t2v_extreme_sp_parallel.sh
# Image-to-video
bash scripts/wan22/extreme/run_wan22_moe_i2v_extreme_sp_parallel.sh
```

### Option B: Manual Installation

If Docker is not available, install the environment manually:

```bash
# 1. Install LightX2V
git clone https://github.com/ModelTC/LightX2V.git
cd LightX2V
uv pip install -v .

# 2. Install NVFP4 Kernel
pip install scikit_build_core uv
git clone https://github.com/NVIDIA/cutlass.git
cd lightx2v_kernel

MAX_JOBS=$(nproc) CMAKE_BUILD_PARALLEL_LEVEL=$(nproc) \
uv build --wheel \
  -Cbuild-dir=build . \
  -Ccmake.define.CUTLASS_PATH=/path/to/cutlass \
  --verbose --color=always --no-build-isolation

pip install dist/*whl --force-reinstall --no-deps

# 3. Run single-GPU inference
# Text-to-video
bash scripts/wan22/extreme/run_wan22_moe_t2v_extreme.sh
# Image-to-video
bash scripts/wan22/extreme/run_wan22_moe_i2v_extreme.sh

# 4. Run multi-GPU sequence-parallel inference
# Text-to-video
bash scripts/wan22/extreme/run_wan22_moe_t2v_extreme_sp_parallel.sh
# Image-to-video
bash scripts/wan22/extreme/run_wan22_moe_i2v_extreme_sp_parallel.sh
```

**Single-GPU Scripts:**

- [run_wan22_moe_t2v_extreme.sh](https://github.com/ModelTC/LightX2V/blob/main/scripts/wan22/extreme/run_wan22_moe_t2v_extreme.sh)
- [run_wan22_moe_i2v_extreme.sh](https://github.com/ModelTC/LightX2V/blob/main/scripts/wan22/extreme/run_wan22_moe_i2v_extreme.sh)

**Multi-GPU Scripts:**

- [run_wan22_moe_t2v_extreme_sp_parallel.sh](https://github.com/ModelTC/LightX2V/blob/main/scripts/wan22/extreme/run_wan22_moe_t2v_extreme_sp_parallel.sh)
- [run_wan22_moe_i2v_extreme_sp_parallel.sh](https://github.com/ModelTC/LightX2V/blob/main/scripts/wan22/extreme/run_wan22_moe_i2v_extreme_sp_parallel.sh)

## 🎬 Generation Results

<div style="background: #f8fafc; border: 1px solid #e2e8f0; border-radius: 8px; padding: 16px; margin: 16px 0;">
<p style="font-style: italic; color: #475569; margin: 0; padding: 12px; background: white; border-radius: 6px; border-left: 4px solid #3b82f6;">
"Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage"
</p>
</div>


| Resolution | Wan2.2-T2V-14B | LightWan2.2-A14B |
| --- | --- | --- |
| 480p | <video controls style="width: 260px; height: 180px; border-radius: 6px; object-fit: cover;" src="https://cdn-uploads.huggingface.co/production/uploads/658e760cccbc1e2cc78b4258/WTHhrzx7XR4S1Ys_6Kzx4.mp4"></video> | <video controls style="width: 260px; height: 180px; border-radius: 6px; object-fit: cover;" src="https://cdn-uploads.huggingface.co/production/uploads/658e760cccbc1e2cc78b4258/zorpw7gm9At0J2kCmvkDr.mp4"></video> |
| 720p | <video controls style="width: 260px; height: 180px; border-radius: 6px; object-fit: cover;" src="https://cdn-uploads.huggingface.co/production/uploads/658e760cccbc1e2cc78b4258/vkiyKj7CJA-r0yTz7TEum.mp4"></video> | <video controls style="width: 260px; height: 180px; border-radius: 6px; object-fit: cover;" src="https://cdn-uploads.huggingface.co/production/uploads/658e760cccbc1e2cc78b4258/TuECbzvW5jI9NHG6GLvIR.mp4"></video> |


## ⚡ Performance Comparison

**Test Environment**: RTX 5090 Single GPU | LightX2V Framework | End-to-End Latency

| Method | Task | GPU Number | Resolution | NFE | E2E Latency | Speedup |
| --- | :---: | ---: | ---: | ---: | ---: | ---: |
| Wan2.2-T2V-14B | T2V | 1 | 480p | 40 | 734.0s | 1.0x |
| LightWan2.2-A14B | T2V | 1 | 480p | 4 | 9.1s | 80.7x |
| Wan2.2-T2V-14B | T2V | 1 | 720p | 40 | 2668.0s | 1.0x |
| LightWan2.2-A14B | T2V | 1 | 720p | 4 | 22.5s | 118.7x |
| Wan2.2-I2V-14B | I2V | 1 | 480p | 40 | 787.0s | 1.0x |
| LightWan2.2-A14B | I2V | 1 | 480p | 4 | 10.7s | 73.9x |
| Wan2.2-I2V-14B | I2V | 1 | 720p | 40 | 2685.0s | 1.0x |
| LightWan2.2-A14B | I2V | 1 | 720p | 4 | 26.7s | 100.5x |

## ⚠️ Notes

### System Requirements

- **Required Hardware**: NVIDIA RTX 50-series GPUs or other Blackwell architecture GPUs.
- **Recommended Runtime**: `lightx2v/lightx2v:26052801-cu130-5090`.

### Dependencies

- Prepare Wan2.2 T5 / VAE components following the standard LightX2V Wan2.2 model structure.
- For I2V, also prepare the required image encoder components and input image according to the LightX2V Wan2.2 I2V script.
- Use Blackwell + NVFP4 kernels for optimal speed and memory efficiency.

### Performance Tips

- Use the provided extreme inference script for the 4-step high-noise / low-noise expert schedule.
- Sparse attention is most beneficial at higher resolutions where self-attention dominates latency.
- Enable CPU offload only when GPU memory is limited, since offload can reduce throughput.

## 🤝 Community

- **🐛 Issues**: [GitHub Issues](https://github.com/ModelTC/LightX2V/issues)
- **🤗 Models**: [HuggingFace Hub](https://huggingface.co/lightx2v/)
- **📖 Documentation**: [LightX2V Docs](https://github.com/ModelTC/LightX2V)

---

<div align="center">

**If you find this project helpful, please give us a ⭐ on [GitHub](https://github.com/ModelTC/LightX2V)**

For questions or issues, please open an issue on [LightX2V](https://github.com/ModelTC/LightX2V/issues) or contact lvchengtao0319@gmail.com.

</div>