o
    p’×iÙ  ã                   @   sX   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„ deƒƒZdS )é    )Ú	dataclass)ÚListÚUnionNé   )Ú
BaseOutputc                   @   s4   e Zd ZU dZeejeje	e	e
jj  f 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__r   ÚtorchÚTensorÚnpÚndarrayr   ÚPILÚImageÚ__annotations__© r   r   úm/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/pipelines/animatediff/pipeline_output.pyr      s   
 &r   )Údataclassesr   Útypingr   r   Únumpyr   Ú	PIL.Imager   r   Úutilsr   r   r   r   r   r   Ú<module>   s    