biy`ddlmZddlmZmZmZddlZddlZ ddl m Z eGdde Z y)) dataclass)ListOptionalUnionN) BaseOutputceZdZUdZeeejjejfe d<e ee e d<e ee e d<y)IFPipelineOutputa  Output class for Stable Diffusion 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. nsfw_detected (`List[bool]`): List of flags denoting whether the corresponding generated image likely represents "not-safe-for-work" (nsfw) content or a watermark. `None` if safety checking could not be performed. watermark_detected (`List[bool]`): List of flags denoting whether the corresponding generated image likely has a watermark. `None` if safety checking could not be performed. images nsfw_detectedwatermark_detectedN) __name__ __module__ __qualname____doc__rrPILImagenpndarray__annotations__rboolk/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/pipelines/deepfloyd_if/pipeline_output.pyr r sH  $syy'3 44DJ'' d,,rr ) dataclassesrtypingrrrnumpyr PIL.Imagerutilsrr rrrr s2!(( -z- -r