Skip to content

GitLab

Unverified Commit ed167940 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Update README.md

parent 20d91782
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
README.md
View file @ ed167940
...@@ -49,8 +49,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-lsun-church").to(torch_device) ...@@ -49,8 +49,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-lsun-church").to(torch_device)
# 2. Sample gaussian noise # 2. Sample gaussian noise
image = torch.randn( image = torch.randn(
(1, unet.in_channels, unet.resolution, unet.resolution) (1, unet.in_channels, unet.resolution, unet.resolution),
generator=generator, generator=generator,
) )
image = image.to(torch_device) image = image.to(torch_device)
...@@ -101,8 +101,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-celeba-hq").to(torch_device) ...@@ -101,8 +101,8 @@ unet = UNetModel.from_pretrained("fusing/ddpm-celeba-hq").to(torch_device)
# 2. Sample gaussian noise # 2. Sample gaussian noise
image = torch.randn( image = torch.randn(
(1, unet.in_channels, unet.resolution, unet.resolution) (1, unet.in_channels, unet.resolution, unet.resolution),
generator=generator, generator=generator,
) )
image = image.to(torch_device) image = image.to(torch_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment