Skip to content

Commit 1f8f615

Browse files
authored
Merge branch 'main' into sentencetransformers-model-id
Signed-off-by: Jebq <[email protected]>
2 parents 6c1e476 + 5bfb6fb commit 1f8f615

File tree

3 files changed

+30
-2
lines changed

3 files changed

+30
-2
lines changed

CHANGELOG.md

Lines changed: 2 additions & 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
@@ -36,6 +37,7 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
3637
- Replaced usage of `is_datetime_or_timedelta_dtype` with `is_timedelta64_dtype` and `is_datetime64_any_dtype` by @rawwar ([#316](https://github.com/opensearch-project/opensearch-py-ml/pull/316))
3738
- use try-except-else block for handling unexpected exceptions during integration tests by @rawwar([#370](https://github.com/opensearch-project/opensearch-py-ml/pull/370))
3839
- Removed pandas version pin in nox tests by @rawwar ([#368](https://github.com/opensearch-project/opensearch-py-ml/pull/368))
40+
- Switch AL2 to AL2023 agent and DockerHub to ECR images in ml-models.JenkinsFile ([#377](https://github.com/opensearch-project/opensearch-py-ml/pull/377))
3941
- Fix to ensure `model_id` is only required once when saving a model in `sentencetransformermodel.py` in ([#376](https://github.com/opensearch-project/opensearch-py-ml/pull/376))
4042
- Add a `config_output_path` param in `make_model_config_json` in `sentencetransformermodel.py` to chose where the saving location of the model config json file. ([#376](https://github.com/opensearch-project/opensearch-py-ml/pull/376))
4143

jenkins/ml-models.JenkinsFile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@ pipeline {
77
agent
88
{
99
docker {
10-
label 'Jenkins-Agent-AL2-X64-C54xlarge-Docker-Host'
10+
label 'Jenkins-Agent-AL2023-X64-C54xlarge-Docker-Host'
1111
image 'opensearchstaging/ci-runner:release-centos7-clients-v4'
12+
registryUrl 'https://public.ecr.aws/'
1213
alwaysPull true
1314
}
1415
}

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.2": {
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.2": {
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)