From 3a4bac9b57b07805d1f46d96555afc5918417622 Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Wed, 29 Jan 2025 15:12:35 +0100
Subject: [PATCH] revert vison2seq atol

---
 tests/onnxruntime/test_modeling.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
index 9ca404202d..9ea0483e35 100644
--- a/tests/onnxruntime/test_modeling.py
+++ b/tests/onnxruntime/test_modeling.py
@@ -4956,6 +4956,9 @@ class ORTModelForVision2SeqIntegrationTest(ORTModelTestMixin):
 
     GENERATION_LENGTH = 100
 
+    ATOL = 1e-3
+    RTOL = 1e-3
+
     def _get_sample_image(self):
         url = "http://images.cocodataset.org/val2017/000000039769.jpg"
         image = Image.open(requests.get(url, stream=True).raw)