Skip to content

Commit

Permalink
refac/pipeline_output (#9582)
Browse files Browse the repository at this point in the history
  • Loading branch information
yijun-lee authored Oct 9, 2024
1 parent acd6d2c commit ec9e526
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions src/diffusers/pipelines/deepfloyd_if/pipeline_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,17 @@

@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.
"""
Expand Down

0 comments on commit ec9e526

Please sign in to comment.