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

Mark slow test as such

Browse Source
This commit is contained in:
Sylvain Gugger
2022-07-11 12:48:57 -04:00
parent b1b8222d80
commit 5ff6f853d7
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/longt5/test_modeling_longt5.py
View File

@@ -584,6 +584,7 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase
model = LongT5Model.from_pretrained(model_name) model = LongT5Model.from_pretrained(model_name)
self.assertIsNotNone(model) self.assertIsNotNone(model)
@slow
def test_export_to_onnx(self): def test_export_to_onnx(self):
config_and_inputs = self.model_tester.prepare_config_and_inputs() config_and_inputs = self.model_tester.prepare_config_and_inputs()
model = LongT5Model(config_and_inputs[0]).to(torch_device) model = LongT5Model(config_and_inputs[0]).to(torch_device)