TensorFlow tests: having from_pt set to True requires torch to be installed. (#10664)

* TF model exists for Blenderbot 400M * Marian * RAG
2021-03-12 06:16:40 -05:00
parent 543d0549f8
commit 184ef8ecd0
3 changed files with 7 additions and 11 deletions
--- a/tests/test_modeling_tf_marian.py
+++ b/tests/test_modeling_tf_marian.py
@@ -350,7 +350,7 @@ class AbstractMarianIntegrationTest(unittest.TestCase):
    @cached_property
    def model(self):
        warnings.simplefilter("error")
-        model: TFMarianMTModel = TFAutoModelForSeq2SeqLM.from_pretrained(self.model_name, from_pt=True)
+        model: TFMarianMTModel = TFAutoModelForSeq2SeqLM.from_pretrained(self.model_name)
        assert isinstance(model, TFMarianMTModel)
        c = model.config
        self.assertListEqual(c.bad_words_ids, [[c.pad_token_id]])