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

update examples after ruff being updated (#36972)

Browse Source 
* update

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-03-25 18:15:47 +01:00
committed by GitHub
parent a41677a68b
commit 121830ab47
20 changed files with 42 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/tensorflow/translation/run_translation.py
View File

@@ -501,9 +501,9 @@ def main():
# region Set decoder_start_token_id
if model.config.decoder_start_token_id is None and isinstance(tokenizer, (MBartTokenizer, MBartTokenizerFast)):
assert (
data_args.target_lang is not None and data_args.source_lang is not None
), "mBart requires --target_lang and --source_lang"
assert data_args.target_lang is not None and data_args.source_lang is not None, (
"mBart requires --target_lang and --source_lang"
)
if isinstance(tokenizer, MBartTokenizer):
model.config.decoder_start_token_id = tokenizer.lang_code_to_id[data_args.target_lang]
else: