diff --git a/openprompt/pipeline_base.py b/openprompt/pipeline_base.py index 2802755..7807f61 100644 --- a/openprompt/pipeline_base.py +++ b/openprompt/pipeline_base.py @@ -48,7 +48,7 @@ def __init__(self, tokenizer: PreTrainedTokenizer = None, tokenizer_wrapper_class = None, verbalizer: Optional[Verbalizer] = None, - max_seq_length: Optional[str] = 512, + max_seq_length: Optional[int] = 512, batch_size: Optional[int] = 1, shuffle: Optional[bool] = False, teacher_forcing: Optional[bool] = False, diff --git a/tutorial/7_ernie_paddlepaddle/dataloader.py b/tutorial/7_ernie_paddlepaddle/dataloader.py index 04a5b1d..34504c5 100644 --- a/tutorial/7_ernie_paddlepaddle/dataloader.py +++ b/tutorial/7_ernie_paddlepaddle/dataloader.py @@ -59,7 +59,7 @@ def __init__(self, tokenizer = None, tokenizer_wrapper_class = None, verbalizer = None, - max_seq_length: Optional[str] = 512, + max_seq_length: Optional[int] = 512, batch_size: Optional[int] = 1, shuffle: Optional[bool] = False, teacher_forcing: Optional[bool] = False,