Skip to content

GitLab

Commit 5ff6f853 authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Mark slow test as such

parent b1b8222d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/longt5/test_modeling_longt5.py
View file @ 5ff6f853
...@@ -584,6 +584,7 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase ...@@ -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)
......
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