ONNX 模型库
返回模型

说明文档

LightEmbed/sbert-sentence-t5-base-onnx

这是 Sentence Transformers 模型 sentence-transformers/sentence-t5-base 的 ONNX 版本,用于句子嵌入,针对速度和轻量化性能进行了优化。通过使用 onnxruntime 和 tokenizers 而非较重的库(如 sentence-transformers 和 transformers),该版本确保了更小的库体积和更快的执行速度。以下是该模型的详细信息:

  • 基础模型:sentence-transformers/sentence-t5-base
  • 嵌入维度:768
  • 最大序列长度:256
  • 磁盘文件大小:0.41 GB
  • 包含池化层:是

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

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

使用方法 (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/sentence-t5-base')
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/sbert-sentence-t5-base-onnx')
embeddings = model.encode(sentences)
print(embeddings)

引用与作者

Binh Nguyen / binhcode25@gmail.com

LightEmbed/sbert-sentence-t5-base-onnx

作者 LightEmbed

sentence-similarity light-embed
↓ 1 ♥ 0

创建时间: 2024-07-25 22:23:31+00:00

更新时间: 2024-07-25 22:26:48+00:00

在 Hugging Face 上查看

文件 (9)

.gitattributes
README.md
config.json
config_sentence_transformers.json
model.onnx ONNX
special_tokens_map.json
spiece.model
tokenizer.json
tokenizer_config.json