Views
No views yet
npm i @huggingface/transformersXenova/ms-marco-MiniLM-L-12-v2.1import { AutoTokenizer, AutoModelForSequenceClassification } from '@huggingface/transformers';
2
3const model = await AutoModelForSequenceClassification.from_pretrained('Xenova/ms-marco-MiniLM-L-12-v2');
4const tokenizer = await AutoTokenizer.from_pretrained('Xenova/ms-marco-MiniLM-L-12-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.597102165222168, -11.141762733459473 ]
21// unquantized: [ 9.450557708740234, -11.160483360290527 ]onnx).