SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix mt5 test on AMD devices (#38081)

Browse Source
This commit is contained in:
ivarflakstad
2025-05-12 16:59:00 +02:00
committed by GitHub
parent 5c85018072
commit 8ea72d12a2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/mt5/test_modeling_mt5.py
View File

@@ -1121,4 +1121,4 @@ class MT5IntegrationTest(unittest.TestCase):
mtf_score = -(labels.shape[-1] * loss.item()) mtf_score = -(labels.shape[-1] * loss.item())
EXPECTED_SCORE = -84.9127 EXPECTED_SCORE = -84.9127
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) self.assertLess(abs(mtf_score - EXPECTED_SCORE), 2e-4)