How to identify which models will work with transformers.js?

[punkpeye]

Question

I've tried multiple models from MTEB dashboard (e.g. jinaai/jina-embeddings-v3, jinaai/jina-embeddings-v2, dunzhang/stella_en_400M_v5), but none of them work.

It's not clear which models will work?

const generateGteSmallEmbedding = await pipeline(
  'feature-extraction',
  'dunzhang/stella_en_400M_v5',
);

[xenova]

Hi there 👋 Models that are supported by transformers.js contain the "transformers.js" tag on the Hugging Face Hub: https://huggingface.co/models?library=transformers.js&sort=downloads

e.g.,

[donicakraningrat]

How do I get existing model to work in transformers.js? What changes need to be made

[vrajshroff]

I am trying to use this model with transformers.js: pyannote/speaker-diarization-3.0. I am not sure how I can do that.

I see a similar model can be used with it: https://huggingface.co/pyannote/segmentation-3.0/tree/main. Thanks for your help!

[vrajshroff]

@xenova I posted this question here. Would love to hear from you if you have some time. Thank you!

[vishnusureshperumbavoor]

How to add a huggingface model to transformers js?