patrickvonplaten commited on
Commit
a30605d
1 Parent(s): 34e20c9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -39,11 +39,11 @@ model_id = "google/ddpm-cat-256"
39
  ddpm = DDPMPipeline.from_pretrained(model_id) # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference
40
 
41
  # run pipeline in inference (sample random noise and denoise)
42
- image = ddpm()["sample"]
43
 
44
 
45
  # save image
46
- image[0].save("ddpm_generated_image.png")
47
  ```
48
 
49
  For more in-detail information, please have a look at the [official inference example](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/diffusers_intro.ipynb)
 
39
  ddpm = DDPMPipeline.from_pretrained(model_id) # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference
40
 
41
  # run pipeline in inference (sample random noise and denoise)
42
+ image = ddpm().images[0]
43
 
44
 
45
  # save image
46
+ image.save("ddpm_generated_image.png")
47
  ```
48
 
49
  For more in-detail information, please have a look at the [official inference example](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/diffusers_intro.ipynb)