Update README.md

35c4d01d · Patrick von Platen · GitHub · ab9f061b · 35c4d01d
Unverified Commit 35c4d01d authored Jun 10, 2022 by Patrick von Platen Committed by GitHub Jun 10, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

README.md README.md +1 -1

No files found.
--- a/README.md
+++ b/README.md
@@ -96,7 +96,7 @@ noise_scheduler = DDIMScheduler.from_config("fusing/ddpm-celeba-hq")
 unet = UNetModel.from_pretrained("fusing/ddpm-celeba-hq").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_inference_steps = 50