From 9eff5353dd509e3b370197e210feb3b9aad989c9 Mon Sep 17 00:00:00 2001 From: flozi00 Date: Sun, 31 May 2020 14:39:38 +0200 Subject: [PATCH] typo --- .../question_answering/question_answering_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/simpletransformers/question_answering/question_answering_model.py b/simpletransformers/question_answering/question_answering_model.py index 42e473f3..a2f091d5 100755 --- 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),