Views
No views yet
SentenceTransformer(
(0): Transformer({'max_seq_length': 8192, 'do_lower_case': False}) with Transformer model: ModernBertModel
(1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
)pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("redis/langcache-embed-v2")
5# Run inference
6sentences = [
7 'What are some examples of crimes understood as a moral turpitude?',
8 'What are some examples of crimes of moral turpitude?',
9 'What are some examples of crimes understood as a legal aptitude?',
10]
11embeddings = model.encode(sentences)
12print(embeddings.shape)
13# [3, 768]
14
15# Get the similarity scores for the embeddings
16similarities = model.similarity(embeddings, embeddings)
17print(similarities.shape)
18# [3, 3]anchor, positive, negative_1, negative_2, and negative_3| anchor | positive | negative_1 | negative_2 | negative_3 |
|---|---|---|---|---|
Is life really what I make of it? | Life is what you make it? | Is life hardly what I take of it? | Life is not entirely what I make of it. | Is life not what I make of it? |
When you visit a website, can a person running the website see your IP address? | Does every website I visit knows my public ip address? | When you avoid a website, can a person hiding the website see your MAC address? | When you send an email, can the recipient see your physical location? | When you visit a website, a person running the website cannot see your IP address. |
What are some cool features about iOS 10? | What are the best new features of iOS 10? | iOS 10 received criticism for its initial bugs and performance issues, and some users found the redesigned apps less intuitive compared to previous versions. | What are the drawbacks of using Android 14? | iOS 10 was widely criticized for its bugs, removal of beloved features, and generally being a downgrade from previous versions. |
MatryoshkaLoss with these parameters:
1{
2 "loss": "CachedMultipleNegativesRankingLoss",
3 "matryoshka_dims": [768,512,256,128,64],
4 "matryoshka_weights": [1,1,1,1,1],
5 "n_dims_per_step": -1
6}



1@inproceedings{langcache-embed-v1,
2 title = "Advancing Semantic Caching for LLMs with Domain-Specific Embeddings and Synthetic Data",
3 author = "Gill, Cechmanek, Hutcherson, Rajamohan, Agarwal, Gulzar, Singh, Dion",
4 month = "04",
5 year = "2025",
6 url = "https://arxiv.org/abs/2504.02268",
7}1@inproceedings{reimers-2019-sentence-bert,
2 title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
3 author = "Reimers, Nils and Gurevych, Iryna",
4 booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
5 month = "11",
6 year = "2019",
7 publisher = "Association for Computational Linguistics",
8 url = "https://arxiv.org/abs/1908.10084",
9}
10
11@misc{kusupati2024matryoshka,
12 title={Matryoshka Representation Learning},
13 author={Aditya Kusupati and Gantavya Bhatt and Aniket Rege and Matthew Wallingford and Aditya Sinha and Vivek Ramanujan and William Howard-Snyder and Kaifeng Chen and Sham Kakade and Prateek Jain and Ali Farhadi},
14 year={2024},
15 eprint={2205.13147},
16 archivePrefix={arXiv},
17 primaryClass={cs.LG}
18}
19
20@misc{gao2021scaling,
21 title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
22 author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
23 year={2021},
24 eprint={2101.06983},
25 archivePrefix={arXiv},
26 primaryClass={cs.LG}
27}