Skip to content

GitLab

Commit d440e21f authored by Malte Pietsch's avatar Malte Pietsch Committed by Julien Chaumond
Browse files

add mapping of roberta for QA

parent 875c4ae4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/transformers/modeling_auto.py
View file @ d440e21f
...@@ -74,6 +74,7 @@ from .modeling_roberta import ( ...@@ -74,6 +74,7 @@ from .modeling_roberta import (
RobertaForSequenceClassification, RobertaForSequenceClassification,
RobertaForTokenClassification, RobertaForTokenClassification,
RobertaModel, RobertaModel,
RobertaForQuestionAnswering
) )
from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadModel from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadModel
from .modeling_transfo_xl import TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP, TransfoXLLMHeadModel, TransfoXLModel from .modeling_transfo_xl import TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP, TransfoXLLMHeadModel, TransfoXLModel
...@@ -177,6 +178,7 @@ MODEL_FOR_QUESTION_ANSWERING_MAPPING = OrderedDict( ...@@ -177,6 +178,7 @@ MODEL_FOR_QUESTION_ANSWERING_MAPPING = OrderedDict(
[ [
(DistilBertConfig, DistilBertForQuestionAnswering), (DistilBertConfig, DistilBertForQuestionAnswering),
(AlbertConfig, AlbertForQuestionAnswering), (AlbertConfig, AlbertForQuestionAnswering),
(RobertaConfig, RobertaForQuestionAnswering),
(BertConfig, BertForQuestionAnswering), (BertConfig, BertForQuestionAnswering),
(XLNetConfig, XLNetForQuestionAnswering), (XLNetConfig, XLNetForQuestionAnswering),
(XLMConfig, XLMForQuestionAnswering), (XLMConfig, XLMForQuestionAnswering),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment