From 89af440e320d757a7cb67d8ca6d3b8ce710630ce Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 7 Jun 2022 18:55:10 +0200 Subject: [PATCH] check with other device --- tests/test_modeling_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index deae3b0117..ea516226c4 100755 --- 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))):