Patch models (#6326)

* TFAlbertFor{TokenClassification, MultipleChoice} * Patch models * BERT and TF BERT info s * Update check_repo
2020-08-10 10:39:17 -04:00
parent 6028ed92bd
commit b99098abc7
6 changed files with 37 additions and 14 deletions
--- a/tests/test_modeling_tf_bert.py
+++ b/tests/test_modeling_tf_bert.py
@@ -265,6 +265,7 @@ class TFBertModelTest(TFModelTesterMixin, unittest.TestCase):
        (
            TFBertModel,
            TFBertForMaskedLM,
+            TFBertLMHeadModel,
            TFBertForNextSentencePrediction,
            TFBertForPreTraining,
            TFBertForQuestionAnswering,