o
    GÆÏiµ  ã                   @   sH   d dl mZ d dlZd dlZd dlZddlmZ eG dd„ deƒƒZ	dS )é    )Ú	dataclassNé   )Ú
BaseOutputc                   @   s2   e Zd ZU dZejejB eee	j
j
  B ed< dS )Ú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__ÚtorchÚTensorÚnpÚndarrayÚlistÚPILÚImageÚ__annotations__© r   r   úc/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/animatediff/pipeline_output.pyr   
   s   
 $r   )
Údataclassesr   Únumpyr   Ú	PIL.Imager   r   Úutilsr   r   r   r   r   r   Ú<module>   s    