riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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())
EXPECTED_SCORE = -84.9127
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4)
self.assertLess(abs(mtf_score - EXPECTED_SCORE), 2e-4)