"vscode:/vscode.git/clone" did not exist on "215e6804d3bb011faead6ddc8db00cdbe31b6e91"
Unverified Commit ec9e5264 authored by Yijun Lee's avatar Yijun Lee Committed by GitHub
Browse files

refac/pipeline_output (#9582)

parent acd6d2c4
......@@ -9,16 +9,17 @@ from ...utils import BaseOutput
@dataclass
class IFPipelineOutput(BaseOutput):
"""
Args:
r"""
Output class for Stable Diffusion pipelines.
images (`List[PIL.Image.Image]` or `np.ndarray`)
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]`)
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]`)
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.
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment