
    wi                     r    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dS )    )	dataclass)ListUnionN   )
BaseOutputc                   p    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__     /root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/pipelines/animatediff/pipeline_output.pyr	   r	      sH         	 	 %,
Dcio1F,GGHHHHHHr   r	   )dataclassesr   typingr   r   numpyr   	PIL.Imager   r   utilsr   r	   r   r   r   <module>r      s    ! ! ! ! ! !                        I I I I I
 I I I I Ir   