Making TF XLM-like models XLA and AMP compliant (#10211)

* Fix Flaubert and XLM * Remove useless cast * Tiny fix * Tiny fix
2021-02-17 18:02:48 +01:00
parent 83d803ba02
commit fdb2351ebb
4 changed files with 82 additions and 85 deletions
--- a/tests/test_modeling_tf_flaubert.py
+++ b/tests/test_modeling_tf_flaubert.py
@@ -331,14 +331,6 @@ class TFFlaubertModelTest(TFModelTesterMixin, unittest.TestCase):
            model = TFFlaubertModel.from_pretrained(model_name)
            self.assertIsNotNone(model)

-    def test_mixed_precision(self):
-        # TODO JP: Make Flaubert float16 compliant
-        pass
-
-    def test_xla_mode(self):
-        # TODO JP: Make Flaubert XLA compliant
-        pass
-

@require_tf
@require_sentencepiece