
    +h                         S SK Jr  S SKJrJr  S SKrS SKrSSK	J
r
Jr  \ " S S\
5      5       r\" 5       (       a)  S SKr\R                  R                   " S S	\
5      5       rgg)
    )	dataclass)ListUnionN   )
BaseOutputis_flax_availablec                   p    \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   Srg)StableDiffusionXLPipelineOutput
   aC  
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.
images N)__name__
__module____qualname____firstlineno____doc__r   r   PILImagenpndarray__annotations____static_attributes__r       q/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/stable_diffusion_xl/pipeline_output.pyr
   r
   
   s*     $syy'344r   r
   c                   8    \ rS rSr% Sr\R                  \S'   Srg)#FlaxStableDiffusionXLPipelineOutput   z
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   r   N)	r   r   r   r   r   r   r   r   r   r   r   r   r   r      s    	 

r   r   )dataclassesr   typingr   r   numpyr   	PIL.Imager   utilsr   r   r
   flaxstructr   r   r   r   <module>r%      se    !    2 
5j 
5 
5 	[[	j 	 		 r   