diff --git a/kolors/models/tokenization_chatglm.py b/kolors/models/tokenization_chatglm.py index 50e44b0..aa238dc 100755 --- a/kolors/models/tokenization_chatglm.py +++ b/kolors/models/tokenization_chatglm.py @@ -243,6 +243,7 @@ class ChatGLMTokenizer(PreTrainedTokenizer): padding_strategy: PaddingStrategy = PaddingStrategy.DO_NOT_PAD, pad_to_multiple_of: Optional[int] = None, return_attention_mask: Optional[bool] = None, + padding_side: Optional[str] = None, ) -> dict: """ Pad encoded inputs (on left/right and up to predefined length or max length in the batch)