Skip to content

GitLab

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

Update README.md

parent ef4365c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ eee80dd2
...@@ -34,6 +34,7 @@ import numpy as np ...@@ -34,6 +34,7 @@ import numpy as np
generator = torch.Generator() generator = torch.Generator()
generator = generator.manual_seed(6694729458485568) generator = generator.manual_seed(6694729458485568)
torch_device = "cuda" if torch.cuda.is_available() else "cpu"
# 1. Load models # 1. Load models
scheduler = GaussianDDPMScheduler.from_config("fusing/ddpm-lsun-church") scheduler = GaussianDDPMScheduler.from_config("fusing/ddpm-lsun-church")
......
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