patrickvonplaten commited on
Commit
dab25f0
1 Parent(s): 9429f03

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -39,7 +39,7 @@ model_id = "google/ddpm-ema-celebahq-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().image
43
 
44
 
45
  # save image
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