
    +h                         S SK Jr  S SKJrJr  S SK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UnionN   )
BaseOutputc                   p    \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   Srg)FluxPipelineOutput   a  
Output class for Flux image generation pipelines.

Args:
    images (`List[PIL.Image.Image]` or `torch.Tensor` or `np.ndarray`)
        List of denoised PIL images of length `batch_size` or numpy array or torch tensor of shape `(batch_size,
        height, width, num_channels)`. PIL images or numpy array present the denoised images of the diffusion
        pipeline. Torch tensors can represent either the denoised images or the intermediate latents ready to be
        passed to the decoder.
images N)__name__
__module____qualname____firstlineno____doc__r   r   PILImagenpndarray__annotations____static_attributes__r       b/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/flux/pipeline_output.pyr	   r	      s*    	 $syy'344r   r	   c                   V    \ rS rSr% Sr\R                  \S'   \R                  \S'   Srg)FluxPriorReduxPipelineOutput   aC  
Output class for Flux Prior Redux 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.
prompt_embedspooled_prompt_embedsr   N)	r   r   r   r   r   torchTensorr   r   r   r   r   r   r      s     <<,,&r   r   )dataclassesr   typingr   r   numpyr   	PIL.Imager   r   utilsr   r	   r   r   r   r   <module>r&      sM    !      5 5 5 ': ' 'r   