diff --git a/tests/models/test_attention_processor.py b/tests/models/test_attention_processor.py
index f9b5924ca5..fadee4a9e3 100644
--- 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))