Views
No views yet
1import torch
2from diffusers import DiffusionPipeline, ConsistencyDecoderVAE
3
4vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=pipe.torch_dtype)
5pipe = StableDiffusionPipeline.from_pretrained(
6 "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
7).to("cuda")
8
9pipe("horse", generator=torch.manual_seed(0)).images| Original Image | GAN Decoder | Consistency Decoder |
|---|---|---|
![]() | ![]() | ![]() |
![]() | ![]() | ![]() |
![]() | ![]() | ![]() |