Skip to content

GitLab

Unverified Commit d453ea61 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

fix tolerance for a bloom slow test (#17634)

parent 120649bf
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/bloom/test_modeling_bloom.py
View file @ d453ea61
...@@ -723,7 +723,7 @@ class BloomEmbeddingTest(unittest.TestCase): ...@@ -723,7 +723,7 @@ class BloomEmbeddingTest(unittest.TestCase):
} }
if cuda_available: 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: else:
self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=3) self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=3)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment