From 8ea72d12a24ed7b594bfd8e166b7705049f2a7b4 Mon Sep 17 00:00:00 2001 From: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> Date: Mon, 12 May 2025 16:59:00 +0200 Subject: [PATCH] Fix mt5 test on AMD devices (#38081) --- tests/models/mt5/test_modeling_mt5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/mt5/test_modeling_mt5.py b/tests/models/mt5/test_modeling_mt5.py index 09c78b047ab..3d8783b91df 100644 --- a/tests/models/mt5/test_modeling_mt5.py +++ b/tests/models/mt5/test_modeling_mt5.py @@ -1121,4 +1121,4 @@ class MT5IntegrationTest(unittest.TestCase): mtf_score = -(labels.shape[-1] * loss.item()) EXPECTED_SCORE = -84.9127 - self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) + self.assertLess(abs(mtf_score - EXPECTED_SCORE), 2e-4)