SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix led torchscript (#12735)

Browse Source 
* 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
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_led.py
View File

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