merge

f8a9bb6f · Patrick von Platen · 6f88cc92 · 89af440e · f8a9bb6f
Commit f8a9bb6f authored Jun 07, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/test_modeling_utils.py tests/test_modeling_utils.py +1 -0

No files found.
--- 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))):