add doc

Author: williamberman

src/diffusers/models/consistency_decoder_vae.py

@@ -50,6 +50,23 @@ class ConsistencyDecoderVaeOutput(BaseOutput):
 
 
 class ConsistencyDecoderVae(ModelMixin, ConfigMixin):
+    r"""
+    The consistency decoder used with DALL-E 3.
+
+    Examples:
+        ```py
+        >>> import torch
+        >>> from diffusers import DiffusionPipeline, ConsistencyDecoderVae
+
+        >>> vae = ConsistencyDecoderVae.from_pretrained("openai/consistency-decoder", torch_dtype=pipe.torch_dtype)
+        >>> pipe = StableDiffusionPipeline.from_pretrained(
+        ...     "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
+        ... ).to("cuda")
+
+        >>> pipe("horse", generator=torch.manual_seed(0)).images
+        ```
+    """
+
     @register_to_config
     def __init__(self, encoder_args, decoder_args, scaling_factor, block_out_channels, latent_channels):
         super().__init__()