LightEmbed/all-MiniLM-L6-v2-onnx

这是 Sentence Transformers 模型 sentence-transformers/all-MiniLM-L6-v2 (https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) 的 ONNX 版本，用于句子嵌入，针对速度和轻量级性能进行了优化。通过使用 onnxruntime 和 tokenizers 而不是像 sentence-transformers 和 transformers 这样较重的库，该版本确保了更小的库体积和更快的执行速度。以下是该模型的详细信息：

• 基础模型：sentence-transformers/all-MiniLM-L6-v2
• 嵌入维度：384
• 最大序列长度：256
• 磁盘文件大小：0.08 GB
• 包含池化：是

此 ONNX 模型包含原始 sentence transformer 模型中的所有组件： Transformer、Pooling、Normalize

<!--- Describe your model here -->

使用方法 (LightEmbed)

安装 LightEmbed 后，使用此模型变得非常简单：

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('LightEmbed/all-MiniLM-L6-v2-onnx')
embeddings = model.encode(sentences)
print(embeddings)

引用与作者

Binh Nguyen / binhcode25@gmail.com