Skip to content

Commit

Permalink
Remove listed pooling types
Browse files Browse the repository at this point in the history
Signed-off-by: Flavia Beo <[email protected]>
  • Loading branch information
flaviabeo committed Nov 5, 2024
1 parent fdf3c22 commit e9417e2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vllm/model_executor/layers/pooler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
"""

Expand Down

0 comments on commit e9417e2

Please sign in to comment.