Skip to content

GitLab

Unverified Commit 4fca1a1b authored by Huamin Li's avatar Huamin Li Committed by GitHub
Browse files

[easy] fix pre commit error on trunk (#26665) 


Signed-off-by: default avatarHuamin Li <3ericli@gmail.com>
parent a6049be7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/model_executor/models/bert.py
View file @ 4fca1a1b
...@@ -609,8 +609,8 @@ class SPLADESparsePooler(Pooler): ...@@ -609,8 +609,8 @@ class SPLADESparsePooler(Pooler):
def __init__( def __init__(
self, self,
mlm_head: nn.Module, mlm_head: nn.Module,
cls_token_id: Optional[int] = 101, cls_token_id: int | None = 101,
sep_token_id: Optional[int] = 102, sep_token_id: int | None = 102,
pooling: str = "max", pooling: str = "max",
remove_cls_sep: bool = True, remove_cls_sep: bool = True,
): ):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment