From 69c76173faaa3831cc4bc6f19b60b4ac8e9e4473 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 22 Jan 2023 14:31:05 +0200 Subject: [PATCH] fix tests --- tests/pipelines/dit/test_dit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/dit/test_dit.py b/tests/pipelines/dit/test_dit.py index ca2b183aab..972d90b53a 100644 --- 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)