riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-25 07:18:58 +06:00
Code Issues Actions 32 Packages Projects Releases Wiki Activity

set tgt_lang of MBart Tokenizer for summarization (#10205)

Browse Source
This commit is contained in:
Zhang Cheng 2021-02-16 23:39:37 +09:00 committed by GitHub
parent 5c2d66a2f5
commit df1b0fb54d
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/seq2seq/run_seq2seq.py
View File

@ -386,7 +386,7 @@ def main():
# For translation we set the codes of our source and target languages (only useful for mBART, the others will # For translation we set the codes of our source and target languages (only useful for mBART, the others will
# ignore those attributes). # ignore those attributes).
if data_args.task.startswith("translation"): if data_args.task.startswith("translation") or isinstance(tokenizer, (MBartTokenizer, MBartTokenizerFast)):
if data_args.source_lang is not None: if data_args.source_lang is not None:
tokenizer.src_lang = data_args.source_lang tokenizer.src_lang = data_args.source_lang
if data_args.target_lang is not None: if data_args.target_lang is not None: