Text-to-Image
Diffusers
Safetensors
alfredplpl commited on
Commit
1b94a6d
1 Parent(s): b37282a

Update README.md

Browse files

We need to import torch.

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -34,6 +34,7 @@ The model can be loaded with it's base pipeline `segmind/SSD-1B`. Next, the sche
34
 
35
  ```python
36
  from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
 
37
 
38
  unet = UNet2DConditionModel.from_pretrained("latent-consistency/lcm-ssd-1b", torch_dtype=torch.float16, variant="fp16")
39
  pipe = DiffusionPipeline.from_pretrained("segmind/SSD-1B", unet=unet, torch_dtype=torch.float16, variant="fp16")
 
34
 
35
  ```python
36
  from diffusers import UNet2DConditionModel, DiffusionPipeline, LCMScheduler
37
+ import torch
38
 
39
  unet = UNet2DConditionModel.from_pretrained("latent-consistency/lcm-ssd-1b", torch_dtype=torch.float16, variant="fp16")
40
  pipe = DiffusionPipeline.from_pretrained("segmind/SSD-1B", unet=unet, torch_dtype=torch.float16, variant="fp16")