← back to Exo

.typings/mflux/models/z_image/latent_creator/z_image_latent_creator.pyi

14 lines

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

import mlx.core as mx

class ZImageLatentCreator:
    @staticmethod
    def create_noise(seed: int, height: int, width: int) -> mx.array: ...
    @staticmethod
    def pack_latents(latents: mx.array, height: int, width: int) -> mx.array: ...
    @staticmethod
    def unpack_latents(latents: mx.array, height: int, width: int) -> mx.array: ...