Skip to content

GitLab

Unverified Commit 0bab4476 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

relax tol attention conversion test (#3842)

parent 1f020876
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/test_attention_processor.py
View file @ 0bab4476
......@@ -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))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment