Commit
•
9855549
1
Parent(s):
b37282a
Update README.md (#2)
Browse files- Update README.md (1b94a6d740a3b70dcb4436e783a544d60df63b25)
Co-authored-by: Yasunori Ozaki <[email protected]>
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")
|