vllm_omni.diffusion.distributed.autoencoders.autoencoder_kl_ltx2 ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

Bases: AutoencoderKLLTX2Video, DistributedVaeMixin

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

init_distributed()

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

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

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

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

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