Text-to-Image
Diffusers
Safetensors
alfredplpl commited on
Commit
0f2a593
1 Parent(s): 224b180

Update README.md

Browse files

We need to import torch

Files changed (1) hide show
  1. README.md +1 -0
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")