Fixed documentation of consistency decoder (#5768)

* Fixed doc for consistency decoder * Style fix

Fixed documentation of consistency decoder (#5768)
* Fixed doc for consistency decoder * Style fix
98457580 · Pedro Gabriel Gengo Lourenço · GitHub · 3b37488f · 98457580
Unverified Commit 98457580 authored Nov 14, 2023 by Pedro Gabriel Gengo Lourenço Committed by GitHub Nov 14, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/consistency_decoder_vae.py src/diffusers/models/consistency_decoder_vae.py +1 -1

No files found.
--- a/src/diffusers/models/consistency_decoder_vae.py
+++ b/src/diffusers/models/consistency_decoder_vae.py
@@ -56,7 +56,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin):
    Examples:
        ```py
        >>> import torch
-        >>> from diffusers import DiffusionPipeline, ConsistencyDecoderVAE
+        >>> from diffusers import StableDiffusionPipeline, ConsistencyDecoderVAE

        >>> vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
        >>> pipe = StableDiffusionPipeline.from_pretrained(