Views
No views yet
npm i @huggingface/transformersXenova/ms-marco-MiniLM-L-2-v2.1import { AutoTokenizer, AutoModelForSequenceClassification } from '@huggingface/transformers';
2
3const model = await AutoModelForSequenceClassification.from_pretrained('Xenova/ms-marco-MiniLM-L-2-v2');
4const tokenizer = await AutoTokenizer.from_pretrained('Xenova/ms-marco-MiniLM-L-2-v2');
5
6const features = tokenizer(
7 ['How many people live in Berlin?', 'How many people live in Berlin?'],
8 {
9 text_pair: [
10 'Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.',
11 'New York City is famous for the Metropolitan Museum of Art.',
12 ],
13 padding: true,
14 truncation: true,
15 }
16)
17
18const scores = await model(features);
19console.log(scores);
20// quantized: [ 9.063430786132812, -11.72588062286377 ]
21// unquantized: [ 8.843852043151855, -11.74354362487793 ]onnx).