Add some tests to the slow suite #11860

2021-05-25 10:06:06 +02:00
parent afe479adb5
commit db0b2477cc
2 changed files with 5 additions and 0 deletions
--- a/tests/test_modeling_bigbird_pegasus.py
+++ b/tests/test_modeling_bigbird_pegasus.py
@@ -361,9 +361,11 @@ class BigBirdPegasusModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.
        model.generate(**input_dict)
        model.generate(**input_dict, do_sample=True, early_stopping=False, num_return_sequences=3)

+    @slow
    def test_batched_forward_original_full(self):
        self._check_batched_forward(attn_type="original_full")

+    @slow
    def test_batched_forward_block_sparse(self):
        self._check_batched_forward(attn_type="block_sparse", tolerance=1e-1)