[ci] fix 3 remaining slow GPU failures (#4584)

This commit is contained in:
Sam Shleifer
2020-05-25 19:20:50 -04:00
committed by GitHub
parent 365d452d4d
commit b86e42e0ac
4 changed files with 5 additions and 5 deletions

View File

@@ -222,6 +222,6 @@ class TFElectraModelTest(TFModelTesterMixin, unittest.TestCase):
@slow
def test_model_from_pretrained(self):
# for model_name in list(TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
for model_name in ["electra-small-discriminator"]:
for model_name in ["google/electra-small-discriminator"]:
model = TFElectraModel.from_pretrained(model_name)
self.assertIsNotNone(model)