
    +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Jr  \" \5      r\ " S S\5      5       r\ " S S\5      5       rg)	    )	dataclass)ListUnionN)
BaseOutput
get_loggerc                   8    \ rS rSr% Sr\R                  \S'   Srg)CosmosPipelineOutput   a  
Output class for Cosmos any-to-world/video pipelines.

Args:
    frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
        List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
        denoised PIL image sequences of length `num_frames.` It can also be a NumPy array or Torch tensor of shape
        `(batch_size, num_frames, channels, height, width)`.
frames N)	__name__
__module____qualname____firstlineno____doc__torchTensor__annotations____static_attributes__r       d/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/cosmos/pipeline_output.pyr	   r	      s     LLr   r	   c                   p    \ rS rSr% Sr\\\R                  R                     \	R                  4   \S'   Srg)CosmosImagePipelineOutput   aF  
Output class for Cosmos any-to-image 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.
imagesr   N)r   r   r   r   r   r   r   PILImagenpndarrayr   r   r   r   r   r   r      s*     $syy'344r   r   )dataclassesr   typingr   r   numpyr   	PIL.Imager   r   diffusers.utilsr   r   r   loggerr	   r   r   r   r   <module>r&      sY    !     2 
H	 :   
5
 
5 
5r   