Fix a small error in summarization example (#11762)

This commit is contained in:
Tomy Hsieh
2021-05-19 02:38:36 +08:00
committed by GitHub
parent 77f9bd18af
commit eb3e072a3b

View File

@@ -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: