o
    GÆÏiV  ã                   @   sL   d dl mZ d dlZd dlmZ eG dd„ deƒƒZeG dd„ deƒƒZdS )é    )Ú	dataclassN)Ú
BaseOutputc                   @   ó   e Zd ZU dZejed< dS )ÚKandinskyPipelineOutputa¹  
    Output class for kandinsky video 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Ú__annotations__© r   r   úb/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/pipelines/kandinsky5/pipeline_output.pyr      ó   
 
r   c                   @   r   )ÚKandinskyImagePipelineOutputa…  
    Output class for kandinsky image pipelines.

    Args:
        image (`torch.Tensor`, `np.ndarray`, or list[PIL.Image.Image]):
            List of image outputs - It can be a nested list of length `batch_size,` with each sub-list containing
            denoised PIL image. It can also be a NumPy array or Torch tensor of shape `(batch_size, channels, height,
            width)`.
    ÚimageNr   r   r   r   r   r      r   r   )Údataclassesr   r   Údiffusers.utilsr   r   r   r   r   r   r   Ú<module>   s    