make style
This commit is contained in:
@@ -264,7 +264,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
|
|||||||
|
|
||||||
|
|
||||||
class GPT2ModelLanguageGenerationTest(unittest.TestCase):
|
class GPT2ModelLanguageGenerationTest(unittest.TestCase):
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
def test_lm_generate_gpt2(self):
|
def test_lm_generate_gpt2(self):
|
||||||
model = GPT2LMHeadModel.from_pretrained("gpt2")
|
model = GPT2LMHeadModel.from_pretrained("gpt2")
|
||||||
|
|||||||
@@ -215,7 +215,6 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase):
|
|||||||
|
|
||||||
|
|
||||||
class TransfoXLModelLanguageGenerationTest(unittest.TestCase):
|
class TransfoXLModelLanguageGenerationTest(unittest.TestCase):
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
def test_lm_generate_transfo_xl_wt103(self):
|
def test_lm_generate_transfo_xl_wt103(self):
|
||||||
model = TransfoXLLMHeadModel.from_pretrained("transfo-xl-wt103")
|
model = TransfoXLLMHeadModel.from_pretrained("transfo-xl-wt103")
|
||||||
|
|||||||
@@ -400,7 +400,6 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase):
|
|||||||
|
|
||||||
|
|
||||||
class XLMModelLanguageGenerationTest(unittest.TestCase):
|
class XLMModelLanguageGenerationTest(unittest.TestCase):
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
def test_lm_generate_xlm_mlm_en_2048(self):
|
def test_lm_generate_xlm_mlm_en_2048(self):
|
||||||
model = XLMWithLMHeadModel.from_pretrained("xlm-mlm-en-2048")
|
model = XLMWithLMHeadModel.from_pretrained("xlm-mlm-en-2048")
|
||||||
|
|||||||
@@ -514,7 +514,6 @@ class XLNetModelTest(ModelTesterMixin, unittest.TestCase):
|
|||||||
|
|
||||||
|
|
||||||
class XLNetModelLanguageGenerationTest(unittest.TestCase):
|
class XLNetModelLanguageGenerationTest(unittest.TestCase):
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
def test_lm_generate_xlnet_base_cased(self):
|
def test_lm_generate_xlnet_base_cased(self):
|
||||||
model = XLNetLMHeadModel.from_pretrained("xlnet-base-cased")
|
model = XLNetLMHeadModel.from_pretrained("xlnet-base-cased")
|
||||||
|
|||||||
Reference in New Issue
Block a user