Update run_translation_no_trainer.py (#16652)

args.model_name_or_path -> args.config_name fix it

Update run_translation_no_trainer.py (#16652)
args.model_name_or_path -> args.config_name fix it
db3edd05 · Heerak Son · GitHub · b9f12bed · db3edd05
Unverified Commit db3edd05 authored Apr 12, 2022 by Heerak Son Committed by GitHub Apr 12, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/translation/run_translation_no_trainer.py examples/pytorch/translation/run_translation_no_trainer.py +1 -1

No files found.
--- 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: