check with other device

2022-06-07 18:55:10 +02:00 · 2022-06-07 18:55:10 +02:00 · 89af440e32
parent b76eea0412
commit 89af440e32
1 changed files with 1 additions and 0 deletions
--- a/tests/test_modeling_utils.py
+++ b/tests/test_modeling_utils.py
@ -172,6 +172,7 @@ class SamplerTesterMixin(unittest.TestCase):

        # 2. Sample gaussian noise
        image = scheduler.sample_noise((1, model.in_channels, model.resolution, model.resolution), device=torch_device, generator=generator)
+        import ipdb; ipdb.set_trace()

        # 3. Denoise
        for t in reversed(range(len(scheduler))):