Skip to content

Commit ec9e526

Browse files
authored
refac/pipeline_output (#9582)
1 parent acd6d2c commit ec9e526

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

src/diffusers/pipelines/deepfloyd_if/pipeline_output.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -9,16 +9,17 @@
99

1010
@dataclass
1111
class IFPipelineOutput(BaseOutput):
12-
"""
13-
Args:
12+
r"""
1413
Output class for Stable Diffusion pipelines.
15-
images (`List[PIL.Image.Image]` or `np.ndarray`)
14+
15+
Args:
16+
images (`List[PIL.Image.Image]` or `np.ndarray`):
1617
List of denoised PIL images of length `batch_size` or numpy array of shape `(batch_size, height, width,
1718
num_channels)`. PIL images or numpy array present the denoised images of the diffusion pipeline.
18-
nsfw_detected (`List[bool]`)
19+
nsfw_detected (`List[bool]`):
1920
List of flags denoting whether the corresponding generated image likely represents "not-safe-for-work"
2021
(nsfw) content or a watermark. `None` if safety checking could not be performed.
21-
watermark_detected (`List[bool]`)
22+
watermark_detected (`List[bool]`):
2223
List of flags denoting whether the corresponding generated image likely has a watermark. `None` if safety
2324
checking could not be performed.
2425
"""

0 commit comments

Comments
 (0)