google
/

ddpm-ema-bedroom-256

Unconditional Image Generation

Model card Files Files and versions Community

patrickvonplaten commited on Nov 8, 2022

Commit

0647f7b

•

1 Parent(s): f6c8a04

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ model_id = "google/ddpm-ema-bedroom-256"
 ddpm = DDPMPipeline.from_pretrained(model_id)  # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference
 # run pipeline in inference (sample random noise and denoise)
-image = ddpm()["sample"]
 # save image

 ddpm = DDPMPipeline.from_pretrained(model_id)  # you can replace DDPMPipeline with DDIMPipeline or PNDMPipeline for faster inference
 # run pipeline in inference (sample random noise and denoise)
+image = ddpm().images[0]
 # save image