Skip to content

Commit b94135e

Browse files
committed
Updated docstrings
1 parent 6587918 commit b94135e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model2vec/distill/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class PoolingType(str, Enum):
2727
"""
2828
Pooling strategies for embedding creation.
2929
30-
- MEAN: masked mean over all tokens (ignores padding).
30+
- MEAN: masked mean over all tokens.
3131
- LAST: last non-padding token (often EOS, common in decoder-style models).
3232
- FIRST: first token hidden state (position 0). In BERT-style encoders,
3333
this corresponds to the [CLS] token representation.

0 commit comments

Comments
 (0)