Modelpublic

tnraro/distiluse-base-multilingual-cased-v1

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face

0likes5downloads

Model Card

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

diff

+ 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;