Unverified Commit db9279de authored by Funtowicz Morgan's avatar Funtowicz Morgan Committed by GitHub
Browse files

Fix QA models binding for Flaubert, XLNet and XLM. (#3100)


Signed-off-by: default avatarMorgan Funtowicz <morgan@huggingface.co>

Format & quality
Signed-off-by: default avatarMorgan Funtowicz <morgan@huggingface.co>

Again.
Signed-off-by: default avatarMorgan Funtowicz <morgan@huggingface.co>
parent e58b3ec5
...@@ -78,7 +78,7 @@ from .modeling_distilbert import ( ...@@ -78,7 +78,7 @@ from .modeling_distilbert import (
) )
from .modeling_flaubert import ( from .modeling_flaubert import (
FLAUBERT_PRETRAINED_MODEL_ARCHIVE_MAP, FLAUBERT_PRETRAINED_MODEL_ARCHIVE_MAP,
FlaubertForQuestionAnswering, FlaubertForQuestionAnsweringSimple,
FlaubertForSequenceClassification, FlaubertForSequenceClassification,
FlaubertModel, FlaubertModel,
FlaubertWithLMHeadModel, FlaubertWithLMHeadModel,
...@@ -97,7 +97,7 @@ from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadM ...@@ -97,7 +97,7 @@ from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadM
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
from .modeling_xlm import ( from .modeling_xlm import (
XLM_PRETRAINED_MODEL_ARCHIVE_MAP, XLM_PRETRAINED_MODEL_ARCHIVE_MAP,
XLMForQuestionAnswering, XLMForQuestionAnsweringSimple,
XLMForSequenceClassification, XLMForSequenceClassification,
XLMModel, XLMModel,
XLMWithLMHeadModel, XLMWithLMHeadModel,
...@@ -111,7 +111,7 @@ from .modeling_xlm_roberta import ( ...@@ -111,7 +111,7 @@ from .modeling_xlm_roberta import (
) )
from .modeling_xlnet import ( from .modeling_xlnet import (
XLNET_PRETRAINED_MODEL_ARCHIVE_MAP, XLNET_PRETRAINED_MODEL_ARCHIVE_MAP,
XLNetForQuestionAnswering, XLNetForQuestionAnsweringSimple,
XLNetForSequenceClassification, XLNetForSequenceClassification,
XLNetForTokenClassification, XLNetForTokenClassification,
XLNetLMHeadModel, XLNetLMHeadModel,
...@@ -225,9 +225,9 @@ MODEL_FOR_QUESTION_ANSWERING_MAPPING = OrderedDict( ...@@ -225,9 +225,9 @@ MODEL_FOR_QUESTION_ANSWERING_MAPPING = OrderedDict(
(AlbertConfig, AlbertForQuestionAnswering), (AlbertConfig, AlbertForQuestionAnswering),
(RobertaConfig, RobertaForQuestionAnswering), (RobertaConfig, RobertaForQuestionAnswering),
(BertConfig, BertForQuestionAnswering), (BertConfig, BertForQuestionAnswering),
(XLNetConfig, XLNetForQuestionAnswering), (XLNetConfig, XLNetForQuestionAnsweringSimple),
(FlaubertConfig, FlaubertForQuestionAnswering), (FlaubertConfig, FlaubertForQuestionAnsweringSimple),
(XLMConfig, XLMForQuestionAnswering), (XLMConfig, XLMForQuestionAnsweringSimple),
] ]
) )
......
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