Skip to content

GitLab

Commit 89af440e authored by Patrick von Platen's avatar Patrick von Platen
Browse files

check with other device

parent b76eea04
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_utils.py
View file @ 89af440e
...@@ -172,6 +172,7 @@ class SamplerTesterMixin(unittest.TestCase): ...@@ -172,6 +172,7 @@ class SamplerTesterMixin(unittest.TestCase):
# 2. Sample gaussian noise # 2. Sample gaussian noise
image = scheduler.sample_noise((1, model.in_channels, model.resolution, model.resolution), device=torch_device, generator=generator) image = scheduler.sample_noise((1, model.in_channels, model.resolution, model.resolution), device=torch_device, generator=generator)
import ipdb; ipdb.set_trace()
# 3. Denoise # 3. Denoise
for t in reversed(range(len(scheduler))): for t in reversed(range(len(scheduler))):
......
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