Update README.md
#4
by
kchoi
- opened
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(
|
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"
|