vllm_omni.diffusion.distributed.autoencoders.autoencoder_kl_hunyuan ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

use_tiling: bool

encode_tile_exec(task: TileTask) -> Tensor

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

encode_tile_split(
    x: Tensor,
) -> tuple[list[TileTask], GridSpec]

from_config(config: Any, **kwargs: Any)

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

spatial_tiled_decode(z: Tensor)

spatial_tiled_encode(x: Tensor)

tile_exec(task: TileTask) -> Tensor

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

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