
    
3j                         S SK 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)    )	dataclassN)
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/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/pipelines/cosmos/pipeline_output.pyr   r      s     LLr   r   c                   l    \ rS rSr% Sr\\R                  R                     \R                  -  \
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   listPILImagenpndarrayr   r   r
   r   r   r   r      s%     !BJJ..r   r   )dataclassesr   numpyr   	PIL.Imager   r   diffusers.utilsr   r   r   loggerr   r   r
   r   r   <module>r$      sV    !    2 
H	 :   
/
 
/ 
/r   