Hugging Face's logo

openai
/
consistency-decoder

Diffusers
Safetensors
stable-diffusion
Model card Files
xet
 Community
5

Update README.md

#2
by sayakpaul HF Staff - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -17,7 +17,7 @@ To original code repository can be found [here](https://github.com/openai/consis
17
  import torch
18
  from diffusers import DiffusionPipeline, ConsistencyDecoderVAE
19
 
20
- vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=pipe.torch_dtype)
21
  pipe = StableDiffusionPipeline.from_pretrained(
22
  "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
23
  ).to("cuda")
 
17
  import torch
18
  from diffusers import DiffusionPipeline, ConsistencyDecoderVAE
19
 
20
+ vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
21
  pipe = StableDiffusionPipeline.from_pretrained(
22
  "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
23
  ).to("cuda")