openai
/

consistency-decoder

stable-diffusion

Model card Files Files and versions Community

Update README.md

#2

by sayakpaul HF staff - opened Nov 10, 2023

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ To original code repository can be found [here](https://github.com/openai/consis
 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")

 import torch
 from diffusers import DiffusionPipeline, ConsistencyDecoderVAE
+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")