o
    p’×iØ  ã                   @   s‚   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mZ eG dd„ deƒƒZeƒ r?d dlZejjG dd	„ d	eƒƒZdS dS )
é    )Ú	dataclass)ÚListÚOptionalÚUnionNé   )Ú
BaseOutputÚis_flax_availablec                   @   s<   e Zd ZU dZeeejj ej	f e
d< eee  e
d< dS )ÚStableDiffusionPipelineOutputaê  
    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)`.
        nsfw_content_detected (`List[bool]`)
            List indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content or
            `None` if safety checking could not be performed.
    ÚimagesÚnsfw_content_detectedN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚPILÚImageÚnpÚndarrayÚ__annotations__r   Úbool© r   r   úr/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/pipelines/stable_diffusion/pipeline_output.pyr	   
   s   
 r	   c                   @   s(   e Zd ZU dZejed< ee ed< dS )Ú!FlaxStableDiffusionPipelineOutputaÇ  
        Output class for Flax-based Stable Diffusion pipelines.

        Args:
            images (`np.ndarray`):
                Denoised images of array shape of `(batch_size, height, width, num_channels)`.
            nsfw_content_detected (`List[bool]`):
                List indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content
                or `None` if safety checking could not be performed.
        r
   r   N)	r   r   r   r   r   r   r   r   r   r   r   r   r   r      s   
 
r   )Údataclassesr   Útypingr   r   r   Únumpyr   Ú	PIL.Imager   Úutilsr   r   r	   ÚflaxÚstructr   r   r   r   r   Ú<module>   s    ü