riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[run_translation.py] fix typo (#11372)

Browse Source 
fix typo

Co-authored-by: johnson <johnson@github.com>
This commit is contained in:
johnson7788 2021-04-22 20:17:11 +08:00 committed by GitHub
parent 58d8795d74
commit 5b5e4ca366
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -377,7 +377,7 @@ def main():
forced_bos_token_id = ( forced_bos_token_id = (
tokenizer.lang_code_to_id[data_args.forced_bos_token] if data_args.forced_bos_token is not None else None tokenizer.lang_code_to_id[data_args.forced_bos_token] if data_args.forced_bos_token is not None else None
) )
model.config.foced_bos_token_id = forced_bos_token_id model.config.forced_bos_token_id = forced_bos_token_id
# Get the language codes for input/target. # Get the language codes for input/target.
source_lang = data_args.source_lang.split("_")[0] source_lang = data_args.source_lang.split("_")[0]