Skip to content

GitLab

"...java/git@developer.sourcefind.cn:Tqrr/opencfd-scu.git" did not exist on "bcdfbfd57c8f3cd6cd65998464bb71a562d49948"
Unverified Commit 77ba494b authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[ConsistencyDecoder] fix: doc type (#5745) 

fix: doc type
parent 1328aeb2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/consistency_decoder_vae.py
View file @ 77ba494b
...@@ -58,7 +58,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin): ...@@ -58,7 +58,7 @@ class ConsistencyDecoderVAE(ModelMixin, ConfigMixin):
>>> import torch >>> import torch
>>> from diffusers import DiffusionPipeline, ConsistencyDecoderVAE >>> 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( >>> pipe = StableDiffusionPipeline.from_pretrained(
... "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16 ... "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
... ).to("cuda") ... ).to("cuda")
......
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