Commit
•
843cb40
1
Parent(s):
224b180
Update README.md (#1)
Browse files- Update README.md (0f2a5933a195ac20cc586010f2491df9243ea40b)
Co-authored-by: Yasunori Ozaki <alfredplpl@users.noreply.huggingface.co>
README.md
CHANGED
|
@@ -35,6 +35,7 @@ Please make sure to either disable `guidance_scale` or use values between 1.0 an
|
|
| 35 |
|
| 36 |
```python
|
| 37 |
from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
|
|
|
|
| 38 |
|
| 39 |
unet = UNet2DConditionModel.from_pretrained("latent-consistency/lcm-sdxl", torch_dtype=torch.float16, variant="fp16")
|
| 40 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", unet=unet, torch_dtype=torch.float16, variant="fp16")
|
|
|
|
| 35 |
|
| 36 |
```python
|
| 37 |
from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
|
| 38 |
+
import torch
|
| 39 |
|
| 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")
|