
    +h                         S SK Jr  S SKJr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OptionalUnionN   )
BaseOutputis_flax_availablec                       \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   \\\      \S'   Srg)StableDiffusionPipelineOutput
   a  
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_detected N)__name__
__module____qualname____firstlineno____doc__r   r   PILImagenpndarray__annotations__r   bool__static_attributes__r       n/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/stable_diffusion/pipeline_output.pyr   r   
   s9    
 $syy'344#DJ//r   r   c                   H    \ rS rSr% Sr\R                  \S'   \\	   \S'   Sr
g)!FlaxStableDiffusionPipelineOutput   a  
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   r   N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s    		 

#Dz)r   r   )dataclassesr   typingr   r   r   numpyr   	PIL.Imager   utilsr   r	   r   flaxstructr   r   r   r   <module>r(      sh    ! ( (   2 0J 0 0" 	[[*J * *	 r   