`.typings/mflux/models/common/vae/vae_tiler.pyi`

28 lines
"""
This type stub file was generated by pyright.
"""

import mlx.core as mx
from typing import Callable

class VAETiler:
    @staticmethod
    def encode_image_tiled(
        *,
        image: mx.array,
        encode_fn: Callable[[mx.array], mx.array],
        latent_channels: int,
        tile_size: tuple[int, int] = ...,
        tile_overlap: tuple[int, int] = ...,
        spatial_scale: int = ...,
    ) -> mx.array: ...
    @staticmethod
    def decode_image_tiled(
        *,
        latent: mx.array,
        decode_fn: Callable[[mx.array], mx.array],
        tile_size: tuple[int, int] = ...,
        tile_overlap: tuple[int, int] = ...,
        spatial_scale: int = ...,
    ) -> mx.array: ...