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 |
```
|