fix tolerance for a bloom slow test (#17634)

huggingface · Jun 14, 2022 · d453ea6 · d453ea6
1 parent 120649b
commit d453ea6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/bloom/test_modeling_bloom.py b/tests/models/bloom/test_modeling_bloom.py
@@ -723,7 +723,7 @@ def test_hidden_states_transformers(self):
         }
 
         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)