ml-cpp/dev-tools/extract_model_ops/validation_models.json at dbf959175f345deece60c929e6ecdb67167ae0f8 · elastic/ml-cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
{
    "bert": "bert-base-uncased",
    "roberta": "roberta-base",
    "distilbert": "distilbert-base-uncased",
    "electra": "google/electra-small-discriminator",
    "mpnet": "microsoft/mpnet-base",
    "deberta": "microsoft/deberta-base",
    "dpr": "facebook/dpr-ctx_encoder-single-nq-base",
    "mobilebert": "google/mobilebert-uncased",
    "xlm-roberta": "xlm-roberta-base",

    "elastic-bge-m3": "elastic/bge-m3",
    "elastic-distilbert-cased-ner": "elastic/distilbert-base-cased-finetuned-conll03-english",
    "elastic-distilbert-uncased-ner": "elastic/distilbert-base-uncased-finetuned-conll03-english",
    "elastic-eis-elser-v2": "elastic/eis-elser-v2",
    "elastic-elser-v2": "elastic/elser-v2",
    "elastic-hugging-face-elser": "elastic/hugging-face-elser",
    "elastic-multilingual-e5-small-optimized": "elastic/multilingual-e5-small-optimized",
    "elastic-splade-v3": "elastic/splade-v3",
    "elastic-test-elser-v2": "elastic/test-elser-v2",

    "elastic-elser-v2-quantized": {"model_id": "elastic/elser-v2", "quantized": true},
    "elastic-eis-elser-v2-quantized": {"model_id": "elastic/eis-elser-v2", "quantized": true},
    "elastic-test-elser-v2-quantized": {"model_id": "elastic/test-elser-v2", "quantized": true},

    "ner-dslim-bert-base": "dslim/bert-base-NER",
    "sentiment-distilbert-sst2": "distilbert-base-uncased-finetuned-sst-2-english",

    "es-multilingual-e5-small": "intfloat/multilingual-e5-small",
    "es-all-MiniLM-L6-v2": "sentence-transformers/all-MiniLM-L6-v2",
    "es-cross-encoder-ms-marco": "cross-encoder/ms-marco-MiniLM-L-6-v2",
    "es-dpr-question-encoder": "facebook/dpr-question_encoder-single-nq-base"
}