diff --git a/tests/test_modeling_ctrl.py b/tests/test_modeling_ctrl.py index 285fba42b9e..c29a0d37235 100644 --- a/tests/test_modeling_ctrl.py +++ b/tests/test_modeling_ctrl.py @@ -215,6 +215,7 @@ class CTRLModelTest(ModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_torch class CTRLModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_ctrl(self): diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index cdc67efee4b..6939e1fe569 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -339,6 +339,7 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_torch class GPT2ModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_gpt2(self): diff --git a/tests/test_modeling_openai.py b/tests/test_modeling_openai.py index 83dd2a95f56..cb8c3dc8190 100644 --- a/tests/test_modeling_openai.py +++ b/tests/test_modeling_openai.py @@ -223,6 +223,7 @@ class OpenAIGPTModelTest(ModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_torch class OPENAIGPTModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_openai_gpt(self): diff --git a/tests/test_modeling_tf_ctrl.py b/tests/test_modeling_tf_ctrl.py index 12a42c73143..a55a52365cc 100644 --- a/tests/test_modeling_tf_ctrl.py +++ b/tests/test_modeling_tf_ctrl.py @@ -205,6 +205,7 @@ class TFCTRLModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFCTRLModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_ctrl(self): diff --git a/tests/test_modeling_tf_gpt2.py b/tests/test_modeling_tf_gpt2.py index b2183e26caf..df2f7c18801 100644 --- a/tests/test_modeling_tf_gpt2.py +++ b/tests/test_modeling_tf_gpt2.py @@ -328,6 +328,7 @@ class TFGPT2ModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFGPT2ModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_gpt2(self): diff --git a/tests/test_modeling_tf_openai_gpt.py b/tests/test_modeling_tf_openai_gpt.py index d5fd21ee7ec..8b91f753a00 100644 --- a/tests/test_modeling_tf_openai_gpt.py +++ b/tests/test_modeling_tf_openai_gpt.py @@ -240,6 +240,7 @@ class TFOpenAIGPTModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFOPENAIGPTModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_openai_gpt(self): diff --git a/tests/test_modeling_tf_transfo_xl.py b/tests/test_modeling_tf_transfo_xl.py index 1d606fd61c8..fd5b71d5937 100644 --- a/tests/test_modeling_tf_transfo_xl.py +++ b/tests/test_modeling_tf_transfo_xl.py @@ -214,6 +214,7 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFTransfoXLModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_transfo_xl_wt103(self): diff --git a/tests/test_modeling_tf_xlm.py b/tests/test_modeling_tf_xlm.py index 261c592edab..a959aae6bc2 100644 --- a/tests/test_modeling_tf_xlm.py +++ b/tests/test_modeling_tf_xlm.py @@ -313,6 +313,7 @@ class TFXLMModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFXLMModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_xlm_mlm_en_2048(self): diff --git a/tests/test_modeling_tf_xlnet.py b/tests/test_modeling_tf_xlnet.py index a0b0ebada71..df198220859 100644 --- a/tests/test_modeling_tf_xlnet.py +++ b/tests/test_modeling_tf_xlnet.py @@ -415,6 +415,7 @@ class TFXLNetModelTest(TFModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_tf class TFXLNetModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_xlnet_base_cased(self): diff --git a/tests/test_modeling_xlm.py b/tests/test_modeling_xlm.py index cb7c6626f4e..35043452085 100644 --- a/tests/test_modeling_xlm.py +++ b/tests/test_modeling_xlm.py @@ -430,6 +430,7 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_torch class XLMModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_xlm_mlm_en_2048(self): diff --git a/tests/test_modeling_xlnet.py b/tests/test_modeling_xlnet.py index 12b3e13f3f9..ef557baf662 100644 --- a/tests/test_modeling_xlnet.py +++ b/tests/test_modeling_xlnet.py @@ -513,6 +513,7 @@ class XLNetModelTest(ModelTesterMixin, unittest.TestCase): self.assertIsNotNone(model) +@require_torch class XLNetModelLanguageGenerationTest(unittest.TestCase): @slow def test_lm_generate_xlnet_base_cased(self):