biddlmZddlmZmZddlZddlZddl Z ddl m Z m Z e e ZeGdde ZeGdde Zy) ) dataclass)ListUnionN) BaseOutput get_loggerc0eZdZUdZej ed<y)CosmosPipelineOutputa Output class for Cosmos any-to-world/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__torchTensor__annotations__e/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/pipelines/cosmos/pipeline_output.pyr r s LLrr cheZdZUdZeeejjejfe d<y)CosmosImagePipelineOutputa^ Output class for Cosmos any-to-image pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL images of length `batch_size` or numpy array of shape `(batch_size, height, width, num_channels)`. PIL images or numpy array present the denoised images of the diffusion pipeline. imagesN) r r r rrrPILImagenpndarrayrrrrrrs* $syy'3 44rr) dataclassesrtypingrrnumpyr PIL.Imagerrdiffusers.utilsrrr loggerr rrrrr"sY! 2 H   :     5 5  5r