From ed167940c5c0b38514863e696146b99d68eed6b0 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 13 Jun 2022 00:15:39 +0200 Subject: [PATCH] Update README.md --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 7d7c8c75..314627c2 100644 --- a/README.md +++ b/README.md @@ -49,8 +49,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-lsun-church").to(torch_device) # 2. Sample gaussian noise image = torch.randn( - (1, unet.in_channels, unet.resolution, unet.resolution) - generator=generator, + (1, unet.in_channels, unet.resolution, unet.resolution), + generator=generator, ) image = image.to(torch_device) @@ -101,8 +101,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-celeba-hq").to(torch_device) # 2. Sample gaussian noise image = torch.randn( - (1, unet.in_channels, unet.resolution, unet.resolution) - generator=generator, + (1, unet.in_channels, unet.resolution, unet.resolution), + generator=generator, ) image = image.to(torch_device)