From 5ff6f853d7d5247924d94e278a4cedc37a5885a4 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Mon, 11 Jul 2022 12:48:57 -0400 Subject: [PATCH] Mark slow test as such --- tests/models/longt5/test_modeling_longt5.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/longt5/test_modeling_longt5.py b/tests/models/longt5/test_modeling_longt5.py index 65375e0faf..61ad68921d 100644 --- a/tests/models/longt5/test_modeling_longt5.py +++ b/tests/models/longt5/test_modeling_longt5.py @@ -584,6 +584,7 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase model = LongT5Model.from_pretrained(model_name) self.assertIsNotNone(model) + @slow def test_export_to_onnx(self): config_and_inputs = self.model_tester.prepare_config_and_inputs() model = LongT5Model(config_and_inputs[0]).to(torch_device)