Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "6edb33558c0c76d136bf3a8091e57621d2c5227c"
Unverified Commit 98457580 authored by Pedro Gabriel Gengo Lourenço's avatar Pedro Gabriel Gengo Lourenço Committed by GitHub
Browse files

Fixed documentation of consistency decoder (#5768) 

* Fixed doc for consistency decoder

* Style fix
parent 3b37488f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/consistency_decoder_vae.py
View file @ 98457580
...@@ -56,7 +56,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin): ...@@ -56,7 +56,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin):
Examples: Examples:
```py ```py
>>> import torch >>> import torch
>>> from diffusers import DiffusionPipeline, ConsistencyDecoderVAE >>> from diffusers import StableDiffusionPipeline, ConsistencyDecoderVAE
>>> vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16) >>> vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
>>> pipe = StableDiffusionPipeline.from_pretrained( >>> pipe = StableDiffusionPipeline.from_pretrained(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment