verl-project/verl-omni
Multimodal RL training framework for diffusion & omni models
What it solves
VeRL-Omni provides a stable and high-performance framework for reinforcement learning (RL) post-training of multimodal generative models. It addresses the specific challenges of multimodal RL—such as complex I/O patterns, compute bottlenecks, and the need for efficient rollout generation—that differ significantly from text-only LLM training.
How it works
The framework is built on top of verl and integrates several optimizations to accelerate the RL pipeline:
- Fast Rollouts: Uses the
vLLM-Omnibackend to optimize generation through rollout routing, batching, and embed caching. - Asynchronous Reward Serving: Supports multiple reward models (e.g., HPSv3, GenRM-OCR) and asynchronous computation to overlap the reward phase with the rollout phase.
- Modular Backends: Offers selectable training backends like VeOmni and FSDP2 with combinable parallelism (USP/TP/DP) for distributed training.
- Stability Mechanisms: Implements rollout correction to skip logP recomputation, ensuring faster and more stable diffusion RL pipelines.
Who it’s for
Researchers and developers working on the post-training alignment of diffusion models (image, video, audio) and omni-modality models that jointly handle text, image, audio, and video.
Highlights
- Broad Model Support: Compatible with Qwen-Image, SD3.5, Wan2.2, LTX2.3, and Qwen3-Omni.
- Diverse Algorithm Support: Implements various RL algorithms including FlowGRPO, DiffusionNFT, Diffusion DPO, and GSPO.
- High Throughput: Achieves approximately 25% higher end-to-end throughput compared to diffusers-based implementations on reference setups.
- Hardware Support: Supports both NVIDIA GPUs and Ascend NPUs.
Related
- Dispatch
- Dispatch
- Project
- Project
- Project