cp-cp commited on
Commit
9a86d34
1 Parent(s): f7fd170

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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
  ```