revert vison2seq atol

huggingface · Jan 29, 2025 · 3a4bac9 · 3a4bac9
1 parent 7c8c56f
commit 3a4bac9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)