Update README.md (#4)
Browse files- Update README.md (51928f2d9c82226f71978037e7a8ffc648efe635)
Co-authored-by: Kilim Choi <[email protected]>
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"
|