
    iV                     d    d dl mZ d dlZd dlmZ e G d de             Ze G d de             Zy)    )	dataclassN)
BaseOutputc                   0    e Zd ZU dZej
                  ed<   y)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__     x/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/pipelines/kandinsky5/pipeline_output.pyr   r      s     LLr   r   c                   0    e Zd ZU dZej
                  ed<   y)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      s     <<r   r   )dataclassesr   r   diffusers.utilsr   r   r   r   r   r   <module>r      sD    !  & j   :  r   