Fix led torchscript (#12735)

* Don't test LED on torchscript

* Typo
This commit is contained in:
Lysandre Debut 2021-07-15 17:48:50 +02:00 committed by GitHub
parent f03580fb02
commit 959d448b3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)