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