From 05a1cb902c4dddcaba7ab3a85f76eba63483307e Mon Sep 17 00:00:00 2001 From: estelleafl Date: Tue, 1 Aug 2023 19:03:29 +0300 Subject: [PATCH] [ldm3d] documentation fixing typos (#4284) * fixed typo * updated doc to be consistent in naming * make style/quality * preprocessing for 4 channels and not 6 * make style * test for 4c * make style/quality * fixed test on cpu * fixed doc typo * changed default ckpt to 4c * Update pipeline_stable_diffusion_ldm3d.py --------- Co-authored-by: Aflalo Co-authored-by: Aflalo Co-authored-by: Aflalo --- .../pipelines/stable_diffusion/ldm3d_diffusion.md | 4 ++-- .../pipeline_stable_diffusion_ldm3d.py | 12 +++++++----- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/docs/source/en/api/pipelines/stable_diffusion/ldm3d_diffusion.md b/docs/source/en/api/pipelines/stable_diffusion/ldm3d_diffusion.md index 141867d289..9d70ab4f88 100644 --- a/docs/source/en/api/pipelines/stable_diffusion/ldm3d_diffusion.md +++ b/docs/source/en/api/pipelines/stable_diffusion/ldm3d_diffusion.md @@ -30,8 +30,8 @@ Make sure to check out the Stable Diffusion [Tips](overview#tips) section to lea - all - __call__ -## StableDiffusionPipelineOutput +## LDM3DPipelineOutput -[[autodoc]] pipelines.stable_diffusion.StableDiffusionPipelineOutput +[[autodoc]] pipelines.stable_diffusion.pipeline_stable_diffusion_ldm3d.LDM3DPipelineOutput - all - __call__ diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py index 7cd33e9536..bf41f08114 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py @@ -41,11 +41,10 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name EXAMPLE_DOC_STRING = """ Examples: - ```py - >>> import torch - >>> from diffusers import StableDiffusionPipeline + ```python + >>> from diffusers import StableDiffusionLDM3DPipeline - >>> pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d") + >>> pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d-4c") >>> pipe = pipe.to("cuda") >>> prompt = "a photo of an astronaut riding a horse on mars" @@ -63,7 +62,10 @@ class LDM3DPipelineOutput(BaseOutput): Output class for Stable Diffusion pipelines. Args: - images (`List[PIL.Image.Image]` or `np.ndarray`) + rgb (`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)`. + depth (`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]`)