
    +h                     d    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g)    )	dataclass)ListUnionN   )
BaseOutputc                       \ rS rSr% Sr\\R                  \R                  \
\
\R                  R                        4   \S'   Srg)AnimateDiffPipelineOutput   a  
 Output class for AnimateDiff 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__r   torchTensornpndarrayr   PILImage__annotations____static_attributes__r       i/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/animatediff/pipeline_output.pyr	   r	      s5    	 %,,

Dciioo1F,GGHHr   r	   )dataclassesr   typingr   r   numpyr   	PIL.Imager   r   utilsr   r	   r   r   r   <module>r!      s5    !      I
 I Ir   