Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

check with other device

This commit is contained in:
Patrick von Platen 2022-06-07 18:55:10 +02:00
parent b76eea0412
commit 89af440e32
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_utils.py
View File

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