o
    GÆÏiè  ã                   @   sn   d dl 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ƒ r5d dl
Z
e
jjG dd„ deƒƒZdS dS )	é    )Ú	dataclassNé   )Ú
BaseOutputÚis_flax_availablec                   @   s(   e Zd ZU dZeejj ejB e	d< dS )ÚStableDiffusionXLPipelineOutputa[  
    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.
    ÚimagesN)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚlistÚPILÚImageÚnpÚndarrayÚ__annotations__© r   r   úk/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/stable_diffusion_xl/pipeline_output.pyr   	   s   
 	r   c                   @   s   e Zd ZU dZejed< dS )Ú#FlaxStableDiffusionXLPipelineOutputzë
        Output class for Flax Stable Diffusion XL pipelines.

        Args:
            images (`np.ndarray`)
                Array of shape `(batch_size, height, width, num_channels)` with images from the diffusion pipeline.
        r   N)r   r	   r
   r   r   r   r   r   r   r   r   r      s   
 r   )Údataclassesr   Únumpyr   Ú	PIL.Imager   Úutilsr   r   r   ÚflaxÚstructr   r   r   r   r   Ú<module>   s    ü