
    +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  \ " S S\5      5       r\ " S S	\5      5       rg)
    )	dataclass)ListOptionalUnionN   )
BaseOutputc                       \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   \\\      \S'   Srg)LEditsPPDiffusionPipelineOutput
   u  
Output class for LEdits++ 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       g/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/ledits_pp/pipeline_output.pyr
   r
   
   s9    
 $syy'344#DJ//r   r
   c                       \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   \\\R                  R                     \	R                  4   \S'   Srg)LEditsPPInversionPipelineOutput   a  
Output class for LEdits++ Diffusion pipelines.

Args:
    input_images (`List[PIL.Image.Image]` or `np.ndarray`)
        List of the cropped and resized input images as PIL images of length `batch_size` or NumPy array of shape `
        (batch_size, height, width, num_channels)`.
    vae_reconstruction_images (`List[PIL.Image.Image]` or `np.ndarray`)
        List of VAE reconstruction of all input images as PIL images of length `batch_size` or NumPy array of shape
        ` (batch_size, height, width, num_channels)`.
r   vae_reconstruction_imagesr   N)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      sK    
 $syy'344$T#))//%:BJJ%FGGr   r   )dataclassesr   typingr   r   r   numpyr   	PIL.Imager   utilsr   r
   r   r   r   r   <module>r&      sP    ! ( (    0j 0 0" Hj H Hr   