From 0bab447670f47c28df60fbd2f6a0f833f75a16f5 Mon Sep 17 00:00:00 2001 From: Will Berman Date: Wed, 21 Jun 2023 12:35:38 -0700 Subject: [PATCH] relax tol attention conversion test (#3842) --- tests/models/test_attention_processor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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))