relax tol attention conversion test (#3842)

0bab4476 · Will Berman · GitHub · 1f020876 · 0bab4476
Unverified Commit 0bab4476 authored Jun 21, 2023 by Will Berman Committed by GitHub Jun 21, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/models/test_attention_processor.py tests/models/test_attention_processor.py +2 -2

No files found.
--- a/tests/models/test_attention_processor.py
+++ b/tests/models/test_attention_processor.py
@@ -115,5 +115,5 @@ class DeprecatedAttentionBlockTests(unittest.TestCase):
            output_type="np",
        ).images

-        self.assertTrue(np.allclose(pre_conversion, conversion))
-        self.assertTrue(np.allclose(conversion, after_conversion))
+        self.assertTrue(np.allclose(pre_conversion, conversion, atol=1e-5))
+        self.assertTrue(np.allclose(conversion, after_conversion, atol=1e-5))