Skip to content

vllm_omni.diffusion.models.ltx2.ops.rope_tables

Cached 3D RoPE tables for LTX DiffVAE kernels.

get_rope_tables

get_rope_tables(
    query: Tensor,
    dim_split: tuple[int, int, int],
    base: float,
) -> tuple[tuple[Tensor, Tensor], ...]