diff --git a/vllm/model_executor/models/bert.py b/vllm/model_executor/models/bert.py index df302aee0bf6b..6e81eb8dc91b3 100644 --- a/vllm/model_executor/models/bert.py +++ b/vllm/model_executor/models/bert.py @@ -609,8 +609,8 @@ class SPLADESparsePooler(Pooler): def __init__( self, mlm_head: nn.Module, - cls_token_id: Optional[int] = 101, - sep_token_id: Optional[int] = 102, + cls_token_id: int | None = 101, + sep_token_id: int | None = 102, pooling: str = "max", remove_cls_sep: bool = True, ):