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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -39,8 +39,7 @@ 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()["sample"]
43
-
44
 
45
  # save image
46
  image[0].save("ddpm_generated_image.png")
 
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")