diff --git a/README.md b/README.md index a6ba944a9e..2009a44ddf 100644 --- a/README.md +++ b/README.md @@ -84,12 +84,12 @@ image_pil.save("test.png") Example: ```python -from modeling_ddpm import DDPM +from diffusers import DiffusionPipeline import PIL.Image import numpy as np # load model and scheduler -ddpm = DDPM.from_pretrained("fusing/ddpm-lsun-bedroom-pipe") +ddpm = DiffusionPipeline.from_pretrained("fusing/ddpm-lsun-bedroom") # run pipeline in inference (sample random noise and denoise) image = ddpm()