SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update run_translation_no_trainer.py (#16652)

Browse Source 
args.model_name_or_path -> args.config_name
fix it
This commit is contained in:
Heerak Son
2022-04-12 21:55:12 +09:00
committed by GitHub
parent b9f12bedd3
commit db3edd050b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/translation/run_translation_no_trainer.py
View File

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