We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6587918 commit b94135eCopy full SHA for b94135e
model2vec/distill/inference.py
@@ -27,7 +27,7 @@ class PoolingType(str, Enum):
27
"""
28
Pooling strategies for embedding creation.
29
30
- - MEAN: masked mean over all tokens (ignores padding).
+ - MEAN: masked mean over all tokens.
31
- LAST: last non-padding token (often EOS, common in decoder-style models).
32
- FIRST: first token hidden state (position 0). In BERT-style encoders,
33
this corresponds to the [CLS] token representation.
0 commit comments