diff --git a/src/transformers/tokenization_bart.py b/src/transformers/tokenization_bart.py index bc346074bd4d..8337c06a6176 100644 --- a/src/transformers/tokenization_bart.py +++ b/src/transformers/tokenization_bart.py @@ -61,7 +61,7 @@ def prepare_seq2seq_batch( max_length: Optional[int] = None, max_target_length: Optional[int] = None, padding: str = "longest", - return_tensors: str = "None", + return_tensors: str = "pt", truncation=True, **kwargs, ) -> BatchEncoding: