o
    GÆÏiu  ã                   @   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                   @   s^   e Zd ZU dZeejj ejB e	d< ee
 dB e	d< eejj ejB dB e	d< edB e	d< dS )Ú!StableDiffusionSafePipelineOutputaö  
    Output class for Safe 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_content_detected (`list[bool]`)
            list of flags denoting whether the corresponding generated image likely represents "not-safe-for-work"
            (nsfw) content, or `None` if safety checking could not be performed.
        images (`list[PIL.Image.Image]` or `np.ndarray`)
            list of denoised PIL images that were flagged by the safety checker any may contain "not-safe-for-work"
            (nsfw) content, or `None` if no safety check was performed or no images were flagged.
        applied_safety_concept (`str`)
            The safety concept that was applied for safety guidance, or `None` if safety guidance was disabled
    ÚimagesNÚnsfw_content_detectedÚunsafe_imagesÚapplied_safety_concept)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚlistÚPILÚImageÚnpÚndarrayÚ__annotations__ÚboolÚstr© r   r   úm/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/stable_diffusion_safe/pipeline_output.pyr      s   
 r   )	Údataclassesr   Únumpyr   Ú	PIL.Imager   Úutilsr   r   r   r   r   r   Ú<module>   s    