Skip to content

Commit 0bab447

Browse files
relax tol attention conversion test (#3842)
1 parent 1f02087 commit 0bab447

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/test_attention_processor.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -115,5 +115,5 @@ def test_conversion_when_using_device_map(self):
115115
output_type="np",
116116
).images
117117

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

0 commit comments

Comments
 (0)