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_xlm.py
+++ b/tests/test_modeling_tf_xlm.py
@@ -327,14 +327,6 @@ class TFXLMModelTest(TFModelTesterMixin, unittest.TestCase):
        config_and_inputs = self.model_tester.prepare_config_and_inputs()
        self.model_tester.create_and_check_xlm_for_multiple_choice(*config_and_inputs)

-    def test_mixed_precision(self):
-        # TODO JP: Make XLM float16 compliant
-        pass
-
-    def test_xla_mode(self):
-        # TODO JP: Make XLM XLA compliant
-        pass
-
    @slow
    def test_model_from_pretrained(self):
        for model_name in TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: