From eb3e072a3b24806c72e35e9246e1cf972de1c77f Mon Sep 17 00:00:00 2001 From: Tomy Hsieh Date: Wed, 19 May 2021 02:38:36 +0800 Subject: [PATCH] Fix a small error in summarization example (#11762) --- examples/pytorch/summarization/run_summarization_no_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/summarization/run_summarization_no_trainer.py b/examples/pytorch/summarization/run_summarization_no_trainer.py index ab204907d4..c25c77d756 100644 --- a/examples/pytorch/summarization/run_summarization_no_trainer.py +++ b/examples/pytorch/summarization/run_summarization_no_trainer.py @@ -338,7 +338,7 @@ def main(): # In distributed training, the .from_pretrained methods guarantee that only one local process can concurrently # download model & vocab. if args.config_name: - config = AutoConfig.from_pretrained(args.model_name_or_path) + config = AutoConfig.from_pretrained(args.config_name) elif args.model_name_or_path: config = AutoConfig.from_pretrained(args.model_name_or_path) else: