fix tests

69c76173 · Patrick von Platen · 926b34b4 · 69c76173
Commit 69c76173 authored Jan 22, 2023 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/pipelines/dit/test_dit.py tests/pipelines/dit/test_dit.py +1 -1

No files found.
--- a/tests/pipelines/dit/test_dit.py
+++ b/tests/pipelines/dit/test_dit.py
@@ -111,7 +111,7 @@ class DiTPipelineIntegrationTests(unittest.TestCase):
            expected_image = load_numpy(
                f"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/dit/{word}.npy"
            )
-            assert np.abs((expected_image - image).abs()) < 1e-3
+            assert np.abs((expected_image - image).max()) < 1e-3
    def test_dit_512_fp16(self):
        generator = torch.manual_seed(0)