diff --git a/vllm/model_executor/layers/pooler.py b/vllm/model_executor/layers/pooler.py index c21f0961d02cc..024badbc17b96 100644 --- a/vllm/model_executor/layers/pooler.py +++ b/vllm/model_executor/layers/pooler.py @@ -28,7 +28,7 @@ class Pooler(nn.Module): 3. Returns structured results as `PoolerOutput`. Attributes: - pooling_type: The type of pooling to use (LAST, ALL, CLS, MEAN). + pooling_type: The type of pooling to use. normalize: Whether to normalize the pooled data. """