Update README.md
Browse files
README.md
CHANGED
|
@@ -8,6 +8,7 @@ python
|
|
| 8 |
from diffusers import DDPMPipeline
|
| 9 |
|
| 10 |
pipeline = DDPMPipeline.from_pretrained('cp-cp/sd-class-butterflies')
|
|
|
|
| 11 |
image = pipeline ().images [0]
|
| 12 |
image
|
| 13 |
```
|
|
|
|
| 8 |
from diffusers import DDPMPipeline
|
| 9 |
|
| 10 |
pipeline = DDPMPipeline.from_pretrained('cp-cp/sd-class-butterflies')
|
| 11 |
+
pipeline.to("cuda")
|
| 12 |
image = pipeline ().images [0]
|
| 13 |
image
|
| 14 |
```
|