Skip to content

Commit 6f603fe

Browse files
authored
adding cross encoder models in list (#378)
* adding cross encoder models in list Signed-off-by: Dhrubo Saha <[email protected]> * added changelog Signed-off-by: Dhrubo Saha <[email protected]> --------- Signed-off-by: Dhrubo Saha <[email protected]>
1 parent e5001df commit 6f603fe

File tree

2 files changed

+27
-1
lines changed

2 files changed

+27
-1
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
1313
- Add support for model connectors by @rawwar in ([#345](https://github.com/opensearch-project/opensearch-py-ml/pull/345))
1414
- Add support for model profiles by @rawwar in ([#358](https://github.com/opensearch-project/opensearch-py-ml/pull/358))
1515
- Support for security default admin credential changes in 2.12.0 in ([#365](https://github.com/opensearch-project/opensearch-py-ml/pull/365))
16+
- adding cross encoder models in the pre-trained traced list ([#378](https://github.com/opensearch-project/opensearch-py-ml/pull/378))
1617

1718

1819
### Changed

utils/model_uploader/model_listing/pretrained_models_all_versions.json

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,5 +146,30 @@
146146
],
147147
"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."
148148
}
149-
},
149+
}
150+
},
151+
{
152+
"name": "huggingface/cross-encoders/ms-marco-MiniLM-L-6-v2",
153+
"versions": {
154+
"1.0.0": {
155+
"format": [
156+
"onnx",
157+
"torch_script"
158+
],
159+
"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."
160+
}
161+
}
162+
},
163+
{
164+
"name": "huggingface/cross-encoders/ms-marco-MiniLM-L-12-v2",
165+
"versions": {
166+
"1.0.0": {
167+
"format": [
168+
"onnx",
169+
"torch_script"
170+
],
171+
"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."
172+
}
173+
}
174+
}
150175
]

0 commit comments

Comments
 (0)