Skip to content

GitLab

Unverified Commit 5b5e4ca3 authored by johnson7788's avatar johnson7788 Committed by GitHub
Browse files

[run_translation.py] fix typo (#11372) 



fix typo
Co-authored-by: default avatarjohnson <johnson@github.com>
parent 58d8795d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/translation/run_translation.py
View file @ 5b5e4ca3
...@@ -377,7 +377,7 @@ def main(): ...@@ -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]
......
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