o Û·i¦ã@snddlmZddlZddlZddlmZmZeGdd„deƒƒZ eƒr5ddl Z e j jGdd„deƒƒZ dSdS) é)Ú dataclassNé)Ú BaseOutputÚis_flax_availablec@s8eZdZUdZeejjejBe d<ee dBe d<dS)ÚStableDiffusionPipelineOutputaê 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)`. nsfw_content_detected (`list[bool]`) list indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content or `None` if safety checking could not be performed. ÚimagesNÚnsfw_content_detected) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚlistÚPILÚImageÚnpÚndarrayÚ__annotations__Úbool©rrúj/home/ubuntu/vllm_env/lib/python3.10/site-packages/diffusers/pipelines/stable_diffusion/pipeline_output.pyr s  rc@s(eZdZUdZejed<eeed<dS)Ú!FlaxStableDiffusionPipelineOutputaÇ Output class for Flax-based Stable Diffusion pipelines. Args: images (`np.ndarray`): Denoised images of array shape of `(batch_size, height, width, num_channels)`. nsfw_content_detected (`list[bool]`): list indicating whether the corresponding generated image contains "not-safe-for-work" (nsfw) content or `None` if safety checking could not be performed. rrN) r r r r rrrr rrrrrrs  r) Ú dataclassesrÚnumpyrÚ PIL.ImagerÚutilsrrrÚflaxÚstructrrrrrÚs ü