openai
/

consistency-decoder

stable-diffusion

Model card Files Files and versions Community

sayakpaul HF staff commited on Nov 10, 2023

Commit

50c1b58

•

1 Parent(s): 63b7a48

Update README.md

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")