Fix a bunch of slow tests (#8634)

* CI should install `sentencepiece`

* Requiring TF

* Fixing some TFDPR bugs

* remove return_dict=False/True hack

Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>
This commit is contained in:
Lysandre Debut
2020-11-19 10:41:41 -05:00
committed by GitHub
parent 5362bb8a6b
commit f2e07e7272
6 changed files with 94 additions and 48 deletions

View File

@@ -340,6 +340,7 @@ class TFBertModelTest(TFModelTesterMixin, unittest.TestCase):
self.assertTrue(layer.split("_")[0] in ["dropout", "classifier"])
@require_tf
class TFBertModelIntegrationTest(unittest.TestCase):
@slow
def test_inference_masked_lm(self):