typo

ThilinaRajapakse · Jun 14, 2020 · 95aa829 · 95aa829
1 parent 4c624a0
commit 95aa829
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simpletransformers/question_answering/question_answering_model.py b/simpletransformers/question_answering/question_answering_model.py
@@ -704,7 +704,7 @@ def evaluate(self, eval_data, output_dir, verbose_logging=False):
                     "token_type_ids": batch[2],
                 }
 
-                if self.args["model_type"] in ["xlm", "roberta", "distilbert", "camembert", "electra", "xlmroberta","bart]:
+                if self.args["model_type"] in ["xlm", "roberta", "distilbert", "camembert", "electra", "xlmroberta","bart"]:
                     del inputs["token_type_ids"]
 
                 example_indices = batch[3]