o
    p’×it  ã                   @   sT   d dl mZ d dlmZmZmZ d dlZd dlZ	ddl
mZ eG dd„ deƒƒZdS )é    )Ú	dataclass)ÚListÚOptionalÚUnionNé   )Ú
BaseOutputc                   @   sL   e Zd ZU dZeeejj ej	f e
d< eee  e
d< eee  e
d< dS )ÚIFPipelineOutputa  
    Args:
    Output class for Stable Diffusion pipelines.
        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.
    ÚimagesÚnsfw_detectedÚwatermark_detectedN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚPILÚImageÚnpÚndarrayÚ__annotations__r   Úbool© r   r   ún/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/pipelines/deepfloyd_if/pipeline_output.pyr   
   s
   
 r   )Údataclassesr   Útypingr   r   r   Únumpyr   Ú	PIL.Imager   Úutilsr   r   r   r   r   r   Ú<module>   s    