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_blenderbot.py
+++ b/tests/test_modeling_tf_blenderbot.py
@@ -309,7 +309,7 @@ class TFBlenderbot400MIntegrationTests(unittest.TestCase):

    @cached_property
    def model(self):
-        model = TFAutoModelForSeq2SeqLM.from_pretrained(self.model_name, from_pt=True)
+        model = TFAutoModelForSeq2SeqLM.from_pretrained(self.model_name)
        return model

    @slow