Fix slow tests v4.2.0 (#9561)

* Fix conversational pipeline test * LayoutLM * ProphetNet * BART * Blenderbot & small * Marian * mBART * Pegasus * Tapas tokenizer * BERT2BERT test * Style * Example requirements * TF BERT2BERT test
2021-01-13 09:55:48 -05:00
parent 04dc65e5c6
commit c949516695
12 changed files with 39 additions and 9 deletions
--- a/tests/test_modeling_tf_bart.py
+++ b/tests/test_modeling_tf_bart.py
@@ -356,6 +356,7 @@ class TFBartHeadTests(unittest.TestCase):


@slow
+@require_tf
 class TFBartModelIntegrationTest(unittest.TestCase):
    def test_inference_no_head(self):
        model = TFBartForConditionalGeneration.from_pretrained("facebook/bart-large").model