[ConsistencyDecoder] fix: doc type (#5745)

fix: doc type

[ConsistencyDecoder] fix: doc type (#5745)
fix: doc type
77ba494b · Sayak Paul · GitHub · 1328aeb2 · 77ba494b
Unverified Commit 77ba494b authored Nov 10, 2023 by Sayak Paul Committed by GitHub Nov 10, 2023
Show 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
@@ -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")