Fix led torchscript (#12735)

* Don't test LED on torchscript * Typo

Fix led torchscript (#12735)
* Don't test LED on torchscript * Typo
959d448b · Lysandre Debut · GitHub · f03580fb · 959d448b
Unverified Commit 959d448b authored Jul 15, 2021 by Lysandre Debut Committed by GitHub Jul 15, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/test_modeling_led.py tests/test_modeling_led.py +1 -0

No files found.
--- a/tests/test_modeling_led.py
+++ b/tests/test_modeling_led.py
@@ -273,6 +273,7 @@ class LEDModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
    is_encoder_decoder = True
    test_pruning = False
    test_missing_keys = False
+    test_torchscript = False

    def setUp(self):
        self.model_tester = LEDModelTester(self)