From ab9f061b968bc1c6bf5a8d302d770432be208fe6 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 10 Jun 2022 14:52:27 +0200 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index caa9e67f77..d18896c701 100644 --- a/README.md +++ b/README.md @@ -48,7 +48,7 @@ noise_scheduler = GaussianDDPMScheduler.from_config("fusing/ddpm-lsun-church") unet = UNetModel.from_pretrained("fusing/ddpm-lsun-church").to(torch_device) # 2. Sample gaussian noise -image = noise_scheduler.sample_noise((1, model.in_channels, model.resolution, model.resolution), device=torch_device, generator=generator) +image = noise_scheduler.sample_noise((1, unet.in_channels, unet.resolution, unet.resolution), device=torch_device, generator=generator) # 3. Denoise num_prediction_steps = len(noise_scheduler)