Question Answering support for Albert and Roberta in TF (#3812)

* Add TFAlbertForQuestionAnswering * Add TFRobertaForQuestionAnswering * Update TFAutoModel with Roberta/Albert for QA * Clean `super` TF Albert calls
2020-04-17 10:45:30 -04:00
parent f399c00610
commit 6d00033e97
6 changed files with 193 additions and 5 deletions
--- a/src/transformers/init.py
+++ b/src/transformers/init.py
@@ -424,6 +424,7 @@ if is_tf_available():
        TFRobertaForMaskedLM,
        TFRobertaForSequenceClassification,
        TFRobertaForTokenClassification,
+        TFRobertaForQuestionAnswering,
        TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP,
    )

@@ -466,6 +467,7 @@ if is_tf_available():
        TFAlbertModel,
        TFAlbertForMaskedLM,
        TFAlbertForSequenceClassification,
+        TFAlbertForQuestionAnswering,
        TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP,
    )