Skip to content

GitLab

Unverified Commit db3edd05 authored by Heerak Son's avatar Heerak Son Committed by GitHub
Browse files

Update run_translation_no_trainer.py (#16652) 

args.model_name_or_path -> args.config_name
fix it
parent b9f12bed
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/translation/run_translation_no_trainer.py
View file @ db3edd05
...@@ -357,7 +357,7 @@ def main(): ...@@ -357,7 +357,7 @@ def main():
# In distributed training, the .from_pretrained methods guarantee that only one local process can concurrently # In distributed training, the .from_pretrained methods guarantee that only one local process can concurrently
# download model & vocab. # download model & vocab.
if args.config_name: 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: elif args.model_name_or_path:
config = AutoConfig.from_pretrained(args.model_name_or_path) config = AutoConfig.from_pretrained(args.model_name_or_path)
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment