Skip to content

Commit 3a4bac9

Browse files
revert vison2seq atol
1 parent 7c8c56f commit 3a4bac9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/onnxruntime/test_modeling.py

+3
Original file line numberDiff line numberDiff line change
@@ -4956,6 +4956,9 @@ class ORTModelForVision2SeqIntegrationTest(ORTModelTestMixin):
49564956

49574957
GENERATION_LENGTH = 100
49584958

4959+
ATOL = 1e-3
4960+
RTOL = 1e-3
4961+
49594962
def _get_sample_image(self):
49604963
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
49614964
image = Image.open(requests.get(url, stream=True).raw)

0 commit comments

Comments
 (0)