Skip to content

Commit a816fcc

Browse files
hl4750xrushi
authored andcommitted
[easy] fix pre commit error on trunk (vllm-project#26665)
Signed-off-by: Huamin Li <3ericli@gmail.com> Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
1 parent 50e4ba7 commit a816fcc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/models/bert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -609,8 +609,8 @@ class SPLADESparsePooler(Pooler):
609609
def __init__(
610610
self,
611611
mlm_head: nn.Module,
612-
cls_token_id: Optional[int] = 101,
613-
sep_token_id: Optional[int] = 102,
612+
cls_token_id: int | None = 101,
613+
sep_token_id: int | None = 102,
614614
pooling: str = "max",
615615
remove_cls_sep: bool = True,
616616
):

0 commit comments

Comments
 (0)