fix tolerance for a bloom slow test (#17634)

2022-06-14 18:14:12 +02:00
parent 120649bf3a
commit d453ea6120
1 changed files with 1 additions and 1 deletions
--- a/tests/models/bloom/test_modeling_bloom.py
+++ b/tests/models/bloom/test_modeling_bloom.py
@@ -723,7 +723,7 @@ class BloomEmbeddingTest(unittest.TestCase):
        }

        if cuda_available:
-            self.assertEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item())
+            self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=4)
        else:
            self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=3)