From 2dc8cb87341223e86220516951bb4ad84f880b4a Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sun, 29 Sep 2019 19:51:01 -0400 Subject: [PATCH] fix unknown imports (*ForMultipleChoice) in run_multiple_choice --- transformers/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/transformers/__init__.py b/transformers/__init__.py index 39370cb327..5248bc9f1b 100644 --- a/transformers/__init__.py +++ b/transformers/__init__.py @@ -74,14 +74,15 @@ if is_torch_available(): GPT2LMHeadModel, GPT2DoubleHeadsModel, load_tf_weights_in_gpt2, GPT2_PRETRAINED_MODEL_ARCHIVE_MAP) from .modeling_xlnet import (XLNetPreTrainedModel, XLNetModel, XLNetLMHeadModel, - XLNetForSequenceClassification, XLNetForQuestionAnsweringSimple, - XLNetForQuestionAnswering, + XLNetForSequenceClassification, XLNetForMultipleChoice, + XLNetForQuestionAnsweringSimple, XLNetForQuestionAnswering, load_tf_weights_in_xlnet, XLNET_PRETRAINED_MODEL_ARCHIVE_MAP) from .modeling_xlm import (XLMPreTrainedModel , XLMModel, XLMWithLMHeadModel, XLMForSequenceClassification, XLMForQuestionAnswering, XLMForQuestionAnsweringSimple, XLM_PRETRAINED_MODEL_ARCHIVE_MAP) - from .modeling_roberta import (RobertaForMaskedLM, RobertaModel, RobertaForSequenceClassification, + from .modeling_roberta import (RobertaForMaskedLM, RobertaModel, + RobertaForSequenceClassification, RobertaForMultipleChoice, ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP) from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel, DistilBertForSequenceClassification, DistilBertForQuestionAnswering,