-
Notifications
You must be signed in to change notification settings - Fork 485
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add auto-truncation to pretrained models #3756
Conversation
Signed-off-by: Naarcha-AWS <[email protected]>
| `sentence-transformers/all-distilroberta-v1` | 768-dimensional dense vector space. | No | - [model_url](https://artifacts.opensearch.org/models/ml-models/huggingface/sentence-transformers/all-distilroberta-v1/1.0.1/torch_script/sentence-transformers_all-distilroberta-v1-1.0.1-torch_script.zip)<br>- [config_url](https://artifacts.opensearch.org/models/ml-models/huggingface/sentence-transformers/all-distilroberta-v1/1.0.1/torch_script/config.json) | - [model_url](https://artifacts.opensearch.org/models/ml-models/huggingface/sentence-transformers/all-distilroberta-v1/1.0.1/onnx/sentence-transformers_all-distilroberta-v1-1.0.1-onnx.zip)<br>- [config_url](https://artifacts.opensearch.org/models/ml-models/huggingface/sentence-transformers/all-distilroberta-v1/1.0.1/onnx/config.json) | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Auto-truncation
value should be Yes
for sentence-transformers/all-distilroberta-v1
model
Signed-off-by: Naarcha-AWS <[email protected]>
|
||
| **Model name** | **Vector dimensions** | **Torchscript artifact** | **ONNX artifact** | | ||
| **Model name** | **Vector dimensions** | **Auto-truncation** | **Torchscript artifact** | **ONNX artifact** | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we also add max token size for truncation? @dhrubo-os
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Discussed offline. We will not add max token size for now.
* Add auto-truncation to pretrained models Signed-off-by: Naarcha-AWS <[email protected]> * Fix review feedback Signed-off-by: Naarcha-AWS <[email protected]> --------- Signed-off-by: Naarcha-AWS <[email protected]> (cherry picked from commit 08a4128) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
* Add auto-truncation to pretrained models * Fix review feedback --------- (cherry picked from commit 08a4128) Signed-off-by: Naarcha-AWS <[email protected]> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
* Add auto-truncation to pretrained models Signed-off-by: Naarcha-AWS <[email protected]> * Fix review feedback Signed-off-by: Naarcha-AWS <[email protected]> --------- Signed-off-by: Naarcha-AWS <[email protected]>
* Add auto-truncation to pretrained models Signed-off-by: Naarcha-AWS <[email protected]> * Fix review feedback Signed-off-by: Naarcha-AWS <[email protected]> --------- Signed-off-by: Naarcha-AWS <[email protected]>
* Add auto-truncation to pretrained models Signed-off-by: Naarcha-AWS <[email protected]> * Fix review feedback Signed-off-by: Naarcha-AWS <[email protected]> --------- Signed-off-by: Naarcha-AWS <[email protected]>
Fixes #3652
Checklist
For more information on following Developer Certificate of Origin and signing off your commits, please check here.