ONNX 模型库
返回模型

说明文档

本模型是以下模型的量化版本:SamLowe/universal-sentence-encoder-multilingual-3-onnx


语言:

  • en
  • ar
  • zh
  • fr
  • de
  • it
  • ja
  • ko
  • nl
  • pl
  • pt
  • es
  • th
  • tr
  • ru
    标签:
  • feature-extraction
  • onnx
  • use
  • text-embedding
  • tensorflow-hub
    许可证:apache-2.0
    推理:false
    示例:
  • text: 谢天谢地 ONNX 可用了,速度快多了!

通用句子编码器多语言版 v3

https://tfhub.dev/google/universal-sentence-encoder-multilingual/3 的 ONNX 版本

该模型的原始 TFHub 版本在其他模型中被引用,例如 https://huggingface.co/vprelovac/universal-sentence-encoder-multilingual-3

概述

请参阅 https://tfhub.dev/google/universal-sentence-encoder-multilingual/3 查看概述和许可证详情。

本模型是 TFHub 原始模型的完整精度版本,采用 ONNX 格式。

它使用 ONNXRuntime Extensions 将分词器嵌入到 ONNX 模型中,因此无需单独的分词器,文本可以直接输入到 ONNX 模型中。

后处理(例如池化、归一化)也在 ONNX 模型内实现,因此无需单独的处理步骤。

使用方法

import onnxruntime as ort
from onnxruntime_extensions import get_library_path
from os import cpu_count

sentences = ["hello world"]

def load_onnx_model(model_filepath):
  _options = ort.SessionOptions()
  _options.inter_op_num_threads, _options.intra_op_num_threads = cpu_count(), cpu_count()
  _options.register_custom_ops_library(get_library_path())
  _providers = ["CPUExecutionProvider"]  # 可以使用 ort.get_available_providers()
  return ort.InferenceSession(path_or_bytes=model_filepath, sess_options=_options, providers=_providers)

model = load_onnx_model("filepath_for_model_dot_onnx")

model_outputs = model.run(output_names=["outputs"], input_feed={"inputs": sentences})[0]
print(model_outputs)

WiseIntelligence/universal-sentence-encoder-multilingual-3-onnx-quantized

作者 WiseIntelligence

↓ 0 ♥ 0

创建时间: 2023-10-26 21:03:26+00:00

更新时间: 2023-10-26 21:06:51+00:00

在 Hugging Face 上查看

文件 (3)

.gitattributes
README.md
embedding_model_quantized.onnx ONNX