Skip to content

vllm_omni.diffusion.distributed.autoencoders

Modules:

Name Description
autoencoder_kl
autoencoder_kl_flux2
autoencoder_kl_hunyuan
autoencoder_kl_hunyuan_video_15
autoencoder_kl_ltx2
autoencoder_kl_qwenimage
autoencoder_kl_wan
distributed_vae_executor
wan_decoder_fast_path

Exact fast path for the Wan VAE decoder on a streaming (frame-at-a-time) decode.

wan_decoder_kernels

Fused kernels for the Wan VAE decoder on channels_last_3d activations.

wan_decoder_utils

Utility functions for the streaming Wan decoder fast path.

wan_spatial_shard

Spatially-sharded Wan VAE decode.

wan_vae_fastpath

Fast paths for the diffusers Wan (2.1/2.2) causal video VAE decoder.