From 7abc1d96d114873d9c3c2f1bc81343fb1407cec4 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Thu, 5 Nov 2020 11:42:24 -0500 Subject: [PATCH] no warn (#8329) --- examples/seq2seq/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/seq2seq/utils.py b/examples/seq2seq/utils.py index 6ab72e7ce6..a96535e464 100644 --- a/examples/seq2seq/utils.py +++ b/examples/seq2seq/utils.py @@ -269,7 +269,7 @@ class Seq2SeqDataCollator: ), f"pad_token_id is not defined for ({self.tokenizer.__class__.__name__}), it must be defined." self.data_args = data_args self.tpu_num_cores = tpu_num_cores - self.dataset_kwargs = {"add_prefix_space": isinstance(tokenizer, BartTokenizer)} + self.dataset_kwargs = {"add_prefix_space": True} if isinstance(tokenizer, BartTokenizer) else {} if data_args.src_lang is not None: self.dataset_kwargs["src_lang"] = data_args.src_lang if data_args.tgt_lang is not None: