add mapping of roberta for QA

d440e21f · Malte Pietsch · Julien Chaumond · 875c4ae4 · d440e21f
Commit d440e21f authored Jan 27, 2020 by Malte Pietsch Committed by Julien Chaumond Jan 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/transformers/modeling_auto.py src/transformers/modeling_auto.py +2 -0

No files found.
--- a/src/transformers/modeling_auto.py
+++ b/src/transformers/modeling_auto.py
@@ -74,6 +74,7 @@ from .modeling_roberta import (
    RobertaForSequenceClassification,
    RobertaForTokenClassification,
    RobertaModel,
+    RobertaForQuestionAnswering
 )
 from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadModel
 from .modeling_transfo_xl import TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP, TransfoXLLMHeadModel, TransfoXLModel
@@ -177,6 +178,7 @@ MODEL_FOR_QUESTION_ANSWERING_MAPPING = OrderedDict(
    [
        (DistilBertConfig, DistilBertForQuestionAnswering),
        (AlbertConfig, AlbertForQuestionAnswering),
+        (RobertaConfig, RobertaForQuestionAnswering),
        (BertConfig, BertForQuestionAnswering),
        (XLNetConfig, XLNetForQuestionAnswering),
        (XLMConfig, XLMForQuestionAnswering),