Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
ec9e5264
Unverified
Commit
ec9e5264
authored
Oct 09, 2024
by
Yijun Lee
Committed by
GitHub
Oct 08, 2024
Browse files
refac/pipeline_output (#9582)
parent
acd6d2c4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
src/diffusers/pipelines/deepfloyd_if/pipeline_output.py
src/diffusers/pipelines/deepfloyd_if/pipeline_output.py
+6
-5
No files found.
src/diffusers/pipelines/deepfloyd_if/pipeline_output.py
View file @
ec9e5264
...
@@ -9,16 +9,17 @@ from ...utils import BaseOutput
...
@@ -9,16 +9,17 @@ from ...utils import BaseOutput
@
dataclass
@
dataclass
class
IFPipelineOutput
(
BaseOutput
):
class
IFPipelineOutput
(
BaseOutput
):
"""
r
"""
Args:
Output class for Stable Diffusion pipelines.
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,
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.
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"
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.
(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
List of flags denoting whether the corresponding generated image likely has a watermark. `None` if safety
checking could not be performed.
checking could not be performed.
"""
"""
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment