From 76170d035a541ed3b871459452fdd14de725868f Mon Sep 17 00:00:00 2001 From: chenyu Date: Sat, 28 Feb 2026 11:22:35 -0500 Subject: [PATCH] relax atol for test_xlm_roberta_large (#15066) --- test/models/test_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/models/test_onnx.py b/test/models/test_onnx.py index 691e80de0c..c036ecf51b 100644 --- a/test/models/test_onnx.py +++ b/test/models/test_onnx.py @@ -87,7 +87,7 @@ class TestHuggingFaceOnnxModels(unittest.TestCase): "input_ids": np.random.randint(0, 250002, (1, 11), dtype=np.int64), "attention_mask": np.ones((1, 11), dtype=np.int64), } - self._validate(repo_id, model_file, custom_inputs) + self._validate(repo_id, model_file, custom_inputs, atol=1e-3) if __name__ == "__main__": unittest.main()