From acd87ae02adfd1fc08f37d367fd250f4f0d92003 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Tue, 7 Jun 2022 15:51:48 +0200 Subject: [PATCH] update pipeline example --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()