alfredplpl
commited on
Commit
•
0f2a593
1
Parent(s):
224b180
Update README.md
Browse filesWe need to import torch
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")
|