diff --git a/examples/run_squad.py b/examples/run_squad.py index f2d29fd6..efa83510 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -498,7 +498,7 @@ def main(): # Load a trained model and vocabulary that you have fine-tuned model = model_class.from_pretrained(args.output_dir) - tokenizer = tokenizer_class.from_pretrained(args.output_dir) + tokenizer = tokenizer_class.from_pretrained(args.output_dir, do_lower_case=args.do_lower_case) model.to(args.device)