Skip to content

GitLab

Unverified Commit 959d448b authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Fix led torchscript (#12735) 

* Don't test LED on torchscript

* Typo
parent f03580fb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_led.py
View file @ 959d448b
...@@ -273,6 +273,7 @@ class LEDModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase): ...@@ -273,6 +273,7 @@ class LEDModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
is_encoder_decoder = True is_encoder_decoder = True
test_pruning = False test_pruning = False
test_missing_keys = False test_missing_keys = False
test_torchscript = False
def setUp(self): def setUp(self):
self.model_tester = LEDModelTester(self) self.model_tester = LEDModelTester(self)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment