
    bi                     d    d dl mZ d dlmZmZ d dlZd dlZd dl	Z	ddl
mZ e G d de             Zy)    )	dataclass)ListUnionN   )
BaseOutputc                       e Zd ZU dZeej                  ej                  e	e	e
j                  j                        f   ed<   y)AnimateDiffPipelineOutputa  
     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)`
    framesN)__name__
__module____qualname____doc__r   torchTensornpndarrayr   PILImage__annotations__     j/home/cdr/jupyterlab/.venv/lib/python3.12/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   