o
    Û·iD  ã                   @   s@   d dl mZ d dlZd dlZddlmZ eG dd„ deƒƒZdS )é    )Ú	dataclassNé   )Ú
BaseOutputc                   @   sH   e Zd ZU dZeejj ejB e	d< ee
 dB e	d< ee
 dB e	d< dS )ÚIFPipelineOutputa   
    Output class for Stable Diffusion pipelines.

    Args:
        images (`list[PIL.Image.Image]` or `np.ndarray`):
            list of denoised PIL images of length `batch_size` or numpy array of shape `(batch_size, height, width,
            num_channels)`. PIL images or numpy array present the denoised images of the diffusion pipeline.
        nsfw_detected (`list[bool]`):
            list of flags denoting whether the corresponding generated image likely represents "not-safe-for-work"
            (nsfw) content or a watermark. `None` if safety checking could not be performed.
        watermark_detected (`list[bool]`):
            list of flags denoting whether the corresponding generated image likely has a watermark. `None` if safety
            checking could not be performed.
    ÚimagesNÚnsfw_detectedÚwatermark_detected)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚlistÚPILÚImageÚnpÚndarrayÚ__annotations__Úbool© r   r   úf/home/ubuntu/vllm_env/lib/python3.10/site-packages/diffusers/pipelines/deepfloyd_if/pipeline_output.pyr   	   s
   
 r   )	Údataclassesr   Únumpyr   Ú	PIL.Imager   Úutilsr   r   r   r   r   r   Ú<module>   s    