Skip to content

GitLab

"tests/git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "b28cd3fba0e75f5f15898343399b8f349f38bbb7"
Commit 96164196 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

Merge branch 'main' of https://github.com/huggingface/diffusers into main

parents 3b8f2452 eee80dd2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ 96164196
...@@ -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