tnraro/distiluse-base-multilingual-cased-v1

By tnraro · ↓ 2 · ♥ 0 · Updated 2024-02-01

What's in this repo

This page lists the files in tnraro/distiluse-base-multilingual-cased-v1, including model.onnx, model_quantized.onnx.

2 ONNX models (.onnx files)
11 files, including 6 .json, 2 .onnx, 1 .md, 1 .txt
Last updated 2024-02-01

ONNX files

File	Size	Download
onnx/model.onnx	—	Download
onnx/model_quantized.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

This is ONNX weights that solved #1519 for Xenova/distiluse-base-multilingual-cased-v1.

Maybe the code in transformers.js needs to be modified as below. (If you are familiar with this area, please help)

@xenova/transformers@2.14.3 /src/pipelines.js:1190

+ if (outputs.sentence_embedding && outputs.sentence_embedding.dims.length === 2) {
+   outputs.sentence_embedding.dims[2] = outputs.sentence_embedding.dims[1]
+   outputs.sentence_embedding.dims[1] = 1
+ }
  /** @type {Tensor} */
- let result = outputs.last_hidden_state ?? outputs.logits;
+ let result = outputs.last_hidden_state ?? outputs.logits ?? outputs.sentence_embedding;

Model info

feature-extraction transformers

transformers onnx distilbert feature-extraction license:apache-2.0 text-embeddings-inference endpoints_compatible region:us

↓ 2 ♥ 0

Created: 2024-02-01

Updated: 2024-02-01

View on Hugging Face

Files (11)

.gitattributes

README.md

config.json

onnx/model.onnx ONNX

onnx/model_quantized.onnx ONNX

ort_config.json

quantize_config.json

special_tokens_map.json

tokenizer.json

tokenizer_config.json

vocab.txt

Related models

More from tnraro

No other models by this author.

More feature-extraction models

Browse all feature-extraction models