
    +h                     d    S SK Jr  S SKJrJr  S SKrS SKrS SKrS SK	J
r
  \ " S S\
5      5       rg)    )	dataclass)ListUnionN)
BaseOutputc                       \ rS rSr% Sr\\R                  \R                  \
\
\R                  R                        4   \S'   Srg)AllegroPipelineOutput   a  
Output class for Allegro 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)`.
frames N)__name__
__module____qualname____firstlineno____doc__r   torchTensornpndarrayr   PILImage__annotations____static_attributes__r       e/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/allegro/pipeline_output.pyr   r      s5     %,,

Dciioo1F,GGHHr   r   )dataclassesr   typingr   r   numpyr   r   r   diffusers.utilsr   r   r   r   r   <module>r      s5    !   
  & IJ I Ir   