vllm_omni.diffusion.models.minimax_h3.ops.vae ¶

Install bit-exact eager optimizations for the MiniMax H3 video VAE.

Modules:

Name	Description
`dispatch`	Hardware dispatch for MiniMax H3 VAE operators.
`qk_norm_rope`	Bit-exact fused Q/K RMSNorm and RoPE for the MiniMax H3 video VAE.
`scaled_residual`	Bit-exact scaled residual update for the MiniMax H3 video VAE.

install_h3_vae_optimizations ¶

install_h3_vae_optimizations(
    decoder: Module, *, device: device
) -> bool

Install the operators selected for device once.