diff --git a/sentence_transformers/models/Pooling.py b/sentence_transformers/models/Pooling.py index 92628cd8a..4e24a1c92 100644 --- a/sentence_transformers/models/Pooling.py +++ b/sentence_transformers/models/Pooling.py @@ -36,6 +36,10 @@ class Pooling(nn.Module): `_ and `Text and Code Embeddings by Contrastive Pre-Training `_. + include_prompt: If set to false, the prompt tokens are not + included in the pooling. This is useful for reproducing + work that does not include the prompt tokens in the pooling + like INSTRUCTOR, but otherwise not recommended. """ POOLING_MODES = (