AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

relax tol attention conversion test (#3842)

Browse Source
This commit is contained in:
Will Berman
2023-06-21 12:35:38 -07:00
committed by GitHub
parent 1f02087607
commit 0bab447670
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/test_attention_processor.py
View File

@@ -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))