All MiniLM L6 V2 Onnx

🚀 onnx-models/all-MiniLM-L6-v2-onnx

このモデルは、テキスト埋め込みを生成するためのsentence-transformers/all-MiniLM-L6-v2のONNXポート版です。

🚀 クイックスタート

このモデルを使用するには、light-embedをインストールする必要があります。

pip install -U light-embed

その後、元のモデル名を指定することでモデルを使用できます。

from light_embed import TextEmbedding
sentences = [
    "This is an example sentence",
    "Each sentence is converted"
]

model = TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')
embeddings = model.encode(sentences)
print(embeddings)

または、onnxモデル名を指定することもできます。

from light_embed import TextEmbedding
sentences = [
    "This is an example sentence",
    "Each sentence is converted"
]

model = TextEmbedding('onnx-models/all-MiniLM-L6-v2-onnx')
embeddings = model.encode(sentences)
print(embeddings)

✨ 主な機能

• このモデルは、テキスト埋め込みを生成するために使用できます。
• ONNX形式で提供されているため、高速な推論が可能です。

📦 インストール

このモデルを使用するには、light-embedをインストールする必要があります。

pip install -U light-embed

💻 使用例

基本的な使用法

from light_embed import TextEmbedding
sentences = [
    "This is an example sentence",
    "Each sentence is converted"
]

model = TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')
embeddings = model.encode(sentences)
print(embeddings)

高度な使用法

from light_embed import TextEmbedding
sentences = [
    "This is an example sentence",
    "Each sentence is converted"
]

model = TextEmbedding('onnx-models/all-MiniLM-L6-v2-onnx')
embeddings = model.encode(sentences)
print(embeddings)

📚 ドキュメント

モデルの詳細

📄 ライセンス

著作権者: Binh Nguyen / binhcode25@gmail.com