Skip to content

Commit 7fdb4e3

Browse files
fix: truncation=True is needed
1 parent 927a5e7 commit 7fdb4e3

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

  • torchTextClassifiers/tokenizers

torchTextClassifiers/tokenizers/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ def tokenize(
129129
text,
130130
padding=padding,
131131
return_tensors="pt",
132+
truncation=True,
132133
max_length=self.output_dim,
133134
return_offsets_mapping=return_offsets_mapping,
134135
) # method from PreTrainedTokenizerFast

0 commit comments

Comments
 (0)