[fix] mobilebert had wrong path, causing slow test failure (#5205)

This commit is contained in:
Sam Shleifer
2020-06-22 23:31:36 -04:00
committed by GitHub
parent f5c2a122e3
commit 0d158e38c9
2 changed files with 1 additions and 7 deletions

View File

@@ -454,12 +454,6 @@ class MobileBertModelTest(ModelTesterMixin, unittest.TestCase):
config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_mobilebert_for_token_classification(*config_and_inputs)
@slow
def test_model_from_pretrained(self):
for model_name in MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
model = MobileBertModel.from_pretrained(model_name)
self.assertIsNotNone(model)
def _long_tensor(tok_lst):
return torch.tensor(tok_lst, dtype=torch.long, device=torch_device,)