[Bugfix] Replace PoolingParams.normalize with use_activation (#32243)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2026-01-13 18:45:42 +08:00
parent eb28e8068d
commit 232214b2ae
21 changed files with 68 additions and 70 deletions
--- a/vllm/model_executor/models/bert.py
+++ b/vllm/model_executor/models/bert.py
@@ -116,8 +116,8 @@ class BertPooler(SequencePooler):

        # Use lambdas so that weights are not registered under `self.head`
        self.head = EmbeddingPoolerHead(
-            projector=lambda x: self.dense(x),
            head_dtype=head_dtype,
+            projector=lambda x: self.dense(x),
            activation=LambdaPoolerActivation(self.act_fn),
        )