From 77ba494b297c58b61bb50fd9b8a623c147ead333 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Fri, 10 Nov 2023 14:05:22 +0530 Subject: [PATCH] [ConsistencyDecoder] fix: doc type (#5745) fix: doc type --- 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 9cb7c60bbbc2..63d8763d14b5 100644 --- a/src/diffusers/models/consistency_decoder_vae.py +++ b/src/diffusers/models/consistency_decoder_vae.py @@ -58,7 +58,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin): >>> import torch >>> from diffusers import DiffusionPipeline, ConsistencyDecoderVAE - >>> vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=pipe.torch_dtype) + >>> vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16) >>> pipe = StableDiffusionPipeline.from_pretrained( ... "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16 ... ).to("cuda")