1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
This commit is contained in:
Patrick von Platen
2022-06-07 16:59:48 +00:00

View File

@@ -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))):