"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "bc36c26fa62fce8b89b13d42e2dbdfafeeb102ca"
Unverified Commit 9459efb8 authored by fxmarty's avatar fxmarty Committed by GitHub
Browse files

Add atol for sliding window test (#30303)

atol for sliding window test
parent 3f20877d
...@@ -3925,7 +3925,7 @@ class ModelTesterMixin: ...@@ -3925,7 +3925,7 @@ class ModelTesterMixin:
# Only non-padding tokens are expected to match. # Only non-padding tokens are expected to match.
self.assertTrue( self.assertTrue(
torch.allclose(res_eager[attention_mask == 1], res_sdpa[attention_mask == 1], rtol=1e-3) torch.allclose(res_eager[attention_mask == 1], res_sdpa[attention_mask == 1], rtol=1e-4, atol=1e-4)
) )
@require_flash_attn @require_flash_attn
......
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