patrickvonplaten commited on
Commit
ff056c9
1 Parent(s): 826ce2e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -39,10 +39,10 @@ model_id = "google/ddpm-cifar10-32"
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().image
43
 
44
  # save image
45
- image[0].save("ddpm_generated_image.png")
46
  ```
47
 
48
  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
  # save image
45
+ image.save("ddpm_generated_image.png")
46
  ```
47
 
48
  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)