onnx-models/all-mpnet-base-v2-onnx

这是 sentence-transformers/all-mpnet-base-v2 的 ONNX 移植版本，用于生成文本嵌入。

模型详情

• 嵌入维度：768
• 最大序列长度：384
• 文件大小：0.41 GB
• ONNX 模型包含的模块：Transformer、Pooling、Normalize

<!--- 在此描述您的模型 -->

使用方法

安装 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-mpnet-base-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-mpnet-base-v2-onnx')
embeddings = model.encode(sentences)
print(embeddings)

引用与作者

Binh Nguyen / binhcode25@gmail.com