From de9f39836f779db218e88ad4600b07e37cc6e442 Mon Sep 17 00:00:00 2001 From: Dhrubo Saha Date: Wed, 21 Feb 2024 13:50:16 -0800 Subject: [PATCH] adding cross encoder models in list Signed-off-by: Dhrubo Saha --- .../pretrained_models_all_versions.json | 27 ++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/utils/model_uploader/model_listing/pretrained_models_all_versions.json b/utils/model_uploader/model_listing/pretrained_models_all_versions.json index 08e5bd19..1ac4c9d0 100644 --- a/utils/model_uploader/model_listing/pretrained_models_all_versions.json +++ b/utils/model_uploader/model_listing/pretrained_models_all_versions.json @@ -146,5 +146,30 @@ ], "description": "This is a neural sparse encoding model: It transfers text into sparse vector, and then extract nonzero index and value to entry and weights. It serves only in ingestion and customer should use tokenizer model in query." } - }, + } + }, + { + "name": "huggingface/cross-encoders/ms-marco-MiniLM-L-6-v2", + "versions": { + "1.0.0": { + "format": [ + "onnx", + "torch_script" + ], + "description": "The model can be used for Information Retrieval: Given a query, encode the query will all possible passages (e.g. retrieved with ElasticSearch). Then sort the passages in a decreasing order." + } + } + }, + { + "name": "huggingface/cross-encoders/ms-marco-MiniLM-L-12-v2", + "versions": { + "1.0.0": { + "format": [ + "onnx", + "torch_script" + ], + "description": "The model can be used for Information Retrieval: Given a query, encode the query will all possible passages (e.g. retrieved with ElasticSearch). Then sort the passages in a decreasing order." + } + } + } ] \ No newline at end of file