We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d93f53c commit 7f74350Copy full SHA for 7f74350
vllm/model_executor/models/bert.py
@@ -609,8 +609,8 @@ class SPLADESparsePooler(Pooler):
609
def __init__(
610
self,
611
mlm_head: nn.Module,
612
- cls_token_id: Optional[int] = 101,
613
- sep_token_id: Optional[int] = 102,
+ cls_token_id: int | None = 101,
+ sep_token_id: int | None = 102,
614
pooling: str = "max",
615
remove_cls_sep: bool = True,
616
):
0 commit comments