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 |
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. |