typo

ThilinaRajapakse · May 31, 2020 · 9eff535 · 9eff535
1 parent 651417f
commit 9eff535
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/simpletransformers/question_answering/question_answering_model.py b/simpletransformers/question_answering/question_answering_model.py
@@ -57,7 +57,7 @@
     ElectraTokenizer,
     LongformerConfig,
     LongformerTokenizer,
-    LongformmerForQuestionAnswering,
+    LongformerForQuestionAnswering,
     RobertaConfig,
     RobertaForQuestionAnswering,
     RobertaTokenizer,
@@ -104,7 +104,7 @@ def __init__(self, model_type, model_name, args=None, use_cuda=True, cuda_device
             "bert": (BertConfig, BertForQuestionAnswering, BertTokenizer),
             "distilbert": (DistilBertConfig, DistilBertForQuestionAnswering, DistilBertTokenizer),
             "electra": (ElectraConfig, ElectraForQuestionAnswering, ElectraTokenizer),
-            "longformer": (LongformerConfig, LongformmerForQuestionAnswering, LongformerTokenizer),
+            "longformer": (LongformerConfig, LongformerForQuestionAnswering, LongformerTokenizer),
             "roberta": (RobertaConfig, RobertaForQuestionAnswering, RobertaTokenizer),
             "xlm": (XLMConfig, XLMForQuestionAnswering, XLMTokenizer),
             "xlmroberta": (XLMRobertaConfig, XLMRobertaForQuestionAnswering, XLMRobertaTokenizer),