Skip to content

vllm_omni.diffusion.distributed.autoencoders.autoencoder_kl_ltx2

logger module-attribute

logger = init_logger(__name__)

DistributedAutoencoderKLLTX2Video

Bases: AutoencoderKLLTX2Video, DistributedVaeMixin

from_pretrained classmethod

from_pretrained(*args: Any, **kwargs: Any)

init_distributed

init_distributed()

tile_exec

tile_exec(
    task: TileTask,
    temb: Tensor | None = None,
    causal: bool | None = None,
) -> Tensor

tile_merge

tile_merge(
    coord_tensor_map: dict[tuple[int, ...], Tensor],
    grid_spec: GridSpec,
) -> Tensor

tile_split

tile_split(z: Tensor) -> tuple[list[TileTask], GridSpec]

tiled_decode

tiled_decode(
    z: Tensor,
    temb: Tensor | None = None,
    causal: bool | None = None,
    return_dict: bool = True,
)

LTX2VaeExecutor

Bases: DistributedVaeExecutor

execute

execute(
    z: Tensor,
    operator: DistributedOperator,
    broadcast_result: bool = True,
)