From 3d335f833c8f3bbf2eae53d1f49e403498af5da3 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Wed, 15 Jun 2022 15:59:16 +0200 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 6377e4a0..6956fbd4 100644 --- a/README.md +++ b/README.md @@ -195,11 +195,11 @@ model = UNetModel.from_pretrained(model_id) scheduler = PNDMScheduler() # load model and scheduler -ddpm = PNDM(unet=model, noise_scheduler=scheduler) +pndm = PNDM(unet=model, noise_scheduler=scheduler) # run pipeline in inference (sample random noise and denoise) with torch.no_grad(): - image = ddpm() + image = pndm() # process image to PIL image_processed = image.cpu().permute(0, 2, 3, 1)