Fuzail22/onnx-msmarco-distilbert-cos-v5

By Fuzail22 · ↓ 2 · ♥ 1 · Updated 2024-09-21

What's in this repo

This page lists the files in Fuzail22/onnx-msmarco-distilbert-cos-v5, including model.onnx.

1 ONNX models (.onnx files)
8 files, including 4 .json, 1 .md, 1 .onnx, 1 .txt
Last updated 2024-09-21

ONNX files

File	Size	Download
onnx/model.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

Usage using Transformer.js for feature extraction

import { env, pipeline } from "@xenova/transformers";

env.allowLocalModels = false;

const extractor = await pipeline(
  "feature-extraction",
  "Fuzail22/onnx-msmarco-distilbert-cos-v5",
  {
    quantized: false
  }
);

const extract = async (text: string) => {
    const output = await extractor(text, {
      pooling: "mean",
      normalize: true
    });
    console.log("extraction output:", output);
};

extact("Open Source is great!")

This is Onnx converted version of the original model The Input Sequence length is 512

Converted Using Xenova

Link to original Model sentence-transformers/msmarco-distilbert-cos-v5

Model info

onnx distilbert base_model:sentence-transformers/msmarco-distilbert-cos-v5 base_model:quantized:sentence-transformers/msmarco-distilbert-cos-v5 license:mit region:us

↓ 2 ♥ 1

Created: 2024-09-21

Updated: 2024-09-21

View on Hugging Face

Files (8)

.gitattributes

README.md

config.json

onnx/model.onnx ONNX

special_tokens_map.json

tokenizer.json

tokenizer_config.json

vocab.txt

Related models

More models

No task type declared.