Commit f5b50c6b authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent ec16142e
......@@ -264,7 +264,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
class GPT2ModelLanguageGenerationTest(unittest.TestCase):
@slow
def test_lm_generate_gpt2(self):
model = GPT2LMHeadModel.from_pretrained("gpt2")
......
......@@ -215,7 +215,6 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase):
class TransfoXLModelLanguageGenerationTest(unittest.TestCase):
@slow
def test_lm_generate_transfo_xl_wt103(self):
model = TransfoXLLMHeadModel.from_pretrained("transfo-xl-wt103")
......
......@@ -400,7 +400,6 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase):
class XLMModelLanguageGenerationTest(unittest.TestCase):
@slow
def test_lm_generate_xlm_mlm_en_2048(self):
model = XLMWithLMHeadModel.from_pretrained("xlm-mlm-en-2048")
......
......@@ -514,7 +514,6 @@ class XLNetModelTest(ModelTesterMixin, unittest.TestCase):
class XLNetModelLanguageGenerationTest(unittest.TestCase):
@slow
def test_lm_generate_xlnet_base_cased(self):
model = XLNetLMHeadModel.from_pretrained("xlnet-base-cased")
......
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