From 959d448b3fe3277b506a8f04ea8d74381f0bf73e Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 15 Jul 2021 17:48:50 +0200 Subject: [PATCH] Fix led torchscript (#12735) * Don't test LED on torchscript * Typo --- tests/test_modeling_led.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_led.py b/tests/test_modeling_led.py index e507922762..58323886c7 100644 --- 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)