Skip to content

GitLab

Commit 9e9d2dbc authored by anton-l's avatar anton-l
Browse files

Fix np.abs

parent 8b4371f7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_modeling_utils.py
View file @ 9e9d2dbc
......@@ -708,7 +708,7 @@ class PipelineTesterMixin(unittest.TestCase):
generator = generator.manual_seed(0)
new_image = new_ddpm(generator=generator)["sample"]
assert (image - new_image).abs().sum() < 1e-5, "Models don't give the same forward pass"
assert np.abs(image - new_image).sum() < 1e-5, "Models don't give the same forward pass"
@slow
def test_from_pretrained_hub(self):
......@@ -726,7 +726,7 @@ class PipelineTesterMixin(unittest.TestCase):
generator = generator.manual_seed(0)
new_image = ddpm_from_hub(generator=generator)["sample"]
assert (image - new_image).abs().sum() < 1e-5, "Models don't give the same forward pass"
assert np.abs(image - new_image).sum() < 1e-5, "Models don't give the same forward pass"
@slow
def test_ddpm_cifar10(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment