o pi+@shddlmZddlmZmZmZddlZddlZ ddl m Z eGddde Z eGdd d e Z dS) ) dataclass)ListOptionalUnionN) BaseOutputc@s<eZdZUdZeeejjej fe d<e ee e d<dS)LEditsPPDiffusionPipelineOutputu Output class for LEdits++ 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. imagesnsfw_content_detectedN) __name__ __module__ __qualname____doc__rrPILImagenpndarray__annotations__rboolrrk/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/pipelines/ledits_pp/pipeline_output.pyr s  rc@sFeZdZUdZeeejjej fe d<eeejjej fe d<dS)LEditsPPInversionPipelineOutputa0 Output class for LEdits++ Diffusion pipelines. Args: input_images (`List[PIL.Image.Image]` or `np.ndarray`) List of the cropped and resized input images as PIL images of length `batch_size` or NumPy array of shape ` (batch_size, height, width, num_channels)`. vae_reconstruction_images (`List[PIL.Image.Image]` or `np.ndarray`) List of VAE reconstruction of all input images as PIL images of length `batch_size` or NumPy array of shape ` (batch_size, height, width, num_channels)`. r vae_reconstruction_imagesN) r r r rrrrrrrrrrrrrs  r) dataclassesrtypingrrrnumpyr PIL.Imagerutilsrrrrrrrs