diff --git a/tests/python/relax/test_frontend_onnx.py b/tests/python/relax/test_frontend_onnx.py index 19be3e0b0358..748119f6f990 100644 --- a/tests/python/relax/test_frontend_onnx.py +++ b/tests/python/relax/test_frontend_onnx.py @@ -1226,6 +1226,8 @@ def verify_attention( "mask_index": mask_index, "relative_position_bias": relative_position_bias, }, + # Maximum observed delta from 500 iterations was 2e-4. + atol=1e-3, ) # "present" output should be nullptr when the "past" input isn't included, # but ort requires an output shape to be specified?