From db3edd050baad744329f353d2e9133b8c8a94639 Mon Sep 17 00:00:00 2001 From: Heerak Son <52475378+raki-1203@users.noreply.github.com> Date: Tue, 12 Apr 2022 21:55:12 +0900 Subject: [PATCH] Update run_translation_no_trainer.py (#16652) args.model_name_or_path -> args.config_name fix it --- examples/pytorch/translation/run_translation_no_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/translation/run_translation_no_trainer.py b/examples/pytorch/translation/run_translation_no_trainer.py index f3f7fc1990..024d705194 100644 --- a/examples/pytorch/translation/run_translation_no_trainer.py +++ b/examples/pytorch/translation/run_translation_no_trainer.py @@ -357,7 +357,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: