Skip to content

vllm_omni.diffusion.models.magi2

SandAI MAGI-2 Preview diffusion support.

Modules:

Name Description
attention

Native MAGI-2 packed attention with sinks and Ulysses exchange.

audio_decoder

Native Stable Audio Open decoder used by MAGI-2.

conditioning

Native MAGI-2 prompt conditioning.

configuration_magi2

Native configuration for the released MAGI-2 Preview checkpoint.

layers

Checkpoint-compatible native layers for MAGI-2 Preview.

mh_moe

Native multi-head MoE used by MAGI-2 Preview.

modeling_magi2

Native MAGI-2 Preview diffusion transformer.

parallel

Distributed primitives for the native MAGI-2 preview transformer.

pipeline_magi2

Native vLLM-Omni pipeline for sand-ai/MAGI-2-preview.

preview_data_proxy

Native packing helpers for the MAGI-2 Preview transformer.

sampler_magi2

Native MAGI-2 Preview classifier-free-guidance sampling.

turbo_vae

Native decoder for the TurboVAED checkpoint shipped with MAGI-2.