Revert "Fix it to work with BART (huggingface#6756)"

This reverts commit 94a8802.
fabiocapsouza · Nov 15, 2020 · e28f778 · e28f778
1 parent f7ffb65
commit e28f778
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/question-answering/run_squad.py b/examples/question-answering/run_squad.py
@@ -187,7 +187,7 @@ def train(args, train_dataset, model, tokenizer):
                 "end_positions": batch[4],
             }
 
-            if args.model_type in ["xlm", "roberta", "distilbert", "camembert", "bart"]:
+            if args.model_type in ["xlm", "roberta", "distilbert", "camembert"]:
                 del inputs["token_type_ids"]
 
             if args.model_type in ["xlnet", "xlm"]:
@@ -300,7 +300,7 @@ def evaluate(args, model, tokenizer, prefix=""):
                 "token_type_ids": batch[2],
             }
 
-            if args.model_type in ["xlm", "roberta", "distilbert", "camembert", "bart"]:
+            if args.model_type in ["xlm", "roberta", "distilbert", "camembert"]:
                 del inputs["token_type_ids"]
 
             feature_indices = batch[3]