up (#10771)

2021-03-17 16:15:14 +03:00
parent c83fbc5f2d
commit f20d75a13f
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_prophetnet.py
+++ b/tests/test_modeling_prophetnet.py
@@ -482,7 +482,7 @@ class ProphetNetModelTester:
            torch.allclose(
                outputs_no_mask.last_hidden_state_ngram[0, :5, 0],
                outputs_with_mask.last_hidden_state_ngram[0, :5, 0],
-                atol=1e-3,
+                atol=1e-2,
            )
        )