All Minilm L6 V2 With Attentions

🚀 Qdrant/bm42-all-minilm-l6-v2-attentions

sentence-transformers/all-MiniLM-L6-v2 のONNXポートで、アテンション重みを返すように調整されています。このモデルは、BM42検索 に使用することを目的としています。

🚀 クイックスタート

このモデルは、sentence-transformers/all-MiniLM-L6-v2 のONNXポートで、アテンション重みを返すように調整されています。このモデルは、BM42検索 に使用することを目的としています。

💻 使用例

基本的な使用法

# Here's an example of performing inference using the model with [FastEmbed](https://github.com/qdrant/fastembed).
from fastembed import SparseTextEmbedding

documents = [
    "You should stay, study and sprint.",
    "History can only prepare us to be surprised yet again.",
]

model = SparseTextEmbedding(model_name="Qdrant/bm42-all-minilm-l6-v2-attentions")
embeddings = list(model.embed(documents))

# [
#     SparseEmbedding(values=array([0.26399775, 0.24662513, 0.47077307]),
#                     indices=array([1881538586, 150760872, 1932363795])),
#     SparseEmbedding(values=array(
#         [0.38320042, 0.25453135, 0.18017513, 0.30432631, 0.1373556]),
#                     indices=array([
#                         733618285, 1849833631, 1008800696, 2090661150,
#                         1117393019
#                     ]))
# ]

📄 ライセンス

このモデルは、Apache-2.0ライセンスの下で提供されています。