
    +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                   8    \ rS rSr% Sr\R                  \S'   Srg)HunyuanVideoPipelineOutput   a  
Output class for HunyuanVideo 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       k/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/hunyuan_video/pipeline_output.pyr   r      s     LLr   r   c                       \ rS rSr% Sr\\R                  \R                  \
\
\R                  R                        \
\R                     4   \S'   Srg)#HunyuanVideoFramepackPipelineOutput   a	  
Output class for HunyuanVideo 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)`. Or, a list of torch tensors where each tensor
        corresponds to a latent that decodes to multiple frames.
r
   r   N)r   r   r   r   r   r   r   r   npndarrayr   PILImager   r   r   r   r   r   r      s@    	 %,,

Dciioo1F,GellI[[\\r   r   )dataclassesr   typingr   r   numpyr   	PIL.Imager   r   diffusers.utilsr   r   r   r   r   r   <module>r#      sP    !     &    ]* ] ]r   