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