From 98457580c061a182e2f99cd9fb80f73db386cc59 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pedro=20Gabriel=20Gengo=20Louren=C3=A7o?= Date: Tue, 14 Nov 2023 17:27:26 -0600 Subject: [PATCH] Fixed documentation of consistency decoder (#5768) * Fixed doc for consistency decoder * Style fix --- src/diffusers/models/consistency_decoder_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(