patrickvonplaten commited on
Commit
9517646
1 Parent(s): d6e79ba

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-ema-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)