Hugging Face's logo

latent-consistency
/
lcm-sdxl

Text-to-Image
Diffusers
Safetensors
Model card Files Community
8

Update README.md

#4
by kchoi - opened
base: refs/heads/main
←
from: refs/pr/4
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ import torch
40
  unet = UNet2DConditionModel.from_pretrained("latent-consistency/lcm-sdxl", torch_dtype=torch.float16, variant="fp16")
41
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", unet=unet, torch_dtype=torch.float16, variant="fp16")
42
 
43
- pipe.scheduler = LCMScheduler.from_config(sd_pipe.scheduler.config)
44
  pipe.to("cuda")
45
 
46
  prompt = "a close-up picture of an old man standing in the rain"
 
40
  unet = UNet2DConditionModel.from_pretrained("latent-consistency/lcm-sdxl", torch_dtype=torch.float16, variant="fp16")
41
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", unet=unet, torch_dtype=torch.float16, variant="fp16")
42
 
43
+ pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
44
  pipe.to("cuda")
45
 
46
  prompt = "a close-up picture of an old man standing in the rain"