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_encoder_decoder.py
+++ b/tests/test_modeling_encoder_decoder.py
@@ -822,7 +822,9 @@ class ProphetNetEncoderDecoderModelTest(EncoderDecoderMixin, unittest.TestCase):
        }

    def get_pretrained_model(self):
-        return EncoderDecoderModel.from_encoder_decoder_pretrained("bert-large-uncased", "prophetnet-large-uncased")
+        return EncoderDecoderModel.from_encoder_decoder_pretrained(
+            "bert-large-uncased", "microsoft/prophetnet-large-uncased"
+        )

    def test_encoder_decoder_model_shared_weights(self):
        pass