diff --git a/src/diffusers/models/consistency_decoder_vae.py b/src/diffusers/models/consistency_decoder_vae.py index 63d8763d14..b182733318 100644 --- 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(