Fixed documentation of consistency decoder (huggingface#5768)

* Fixed doc for consistency decoder * Style fix
qAison · Nov 14, 2023 · 9845758 · 9845758
1 parent 3b37488
commit 9845758
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(