diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index d256bcd10e..752659248d 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -582,7 +582,7 @@ class ModelTesterMixin: ) @slow - def test_torchscript(self): + def test_torchscript_simple(self): config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() self._create_and_check_torchscript(config, inputs_dict)