Ensure pad_to_multiple_of is correctly forwarded in base_tokenizer.py

This commit is contained in:
Morgan Funtowicz
2020-05-31 10:02:59 +02:00
parent 0934fe5803
commit fcb4e76d9b

View File

@@ -84,6 +84,7 @@ class BaseTokenizer:
""" """
return self._tokenizer.enable_padding( return self._tokenizer.enable_padding(
direction=direction, direction=direction,
pad_to_multiple_of=pad_to_multiple_of,
pad_id=pad_id, pad_id=pad_id,
pad_type_id=pad_type_id, pad_type_id=pad_type_id,
pad_token=pad_token, pad_token=pad_token,