Skip to content
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

Merged
merged 2 commits into from
Apr 13, 2023
Merged

Add auto-truncation to pretrained models #3756

merged 2 commits into from
Apr 13, 2023

Conversation

Naarcha-AWS
Copy link
Collaborator

Fixes #3652

Checklist

  • By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license and subject to the Developers Certificate of Origin.
    For more information on following Developer Certificate of Origin and signing off your commits, please check here.

@Naarcha-AWS Naarcha-AWS added 3 - Tech review PR: Tech review in progress backport 2.6 PR: Backport label for 2.6 labels Apr 12, 2023
@Naarcha-AWS Naarcha-AWS self-assigned this Apr 12, 2023
| `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) |
Copy link
Contributor

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]>
@Naarcha-AWS Naarcha-AWS added 4 - Doc review PR: Doc review in progress and removed 3 - Tech review PR: Tech review in progress labels Apr 12, 2023

| **Model name** | **Vector dimensions** | **Torchscript artifact** | **ONNX artifact** |
| **Model name** | **Vector dimensions** | **Auto-truncation** | **Torchscript artifact** | **ONNX artifact** |
Copy link
Contributor

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

Copy link
Contributor

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.

@Naarcha-AWS Naarcha-AWS merged commit 08a4128 into main Apr 13, 2023
@Naarcha-AWS Naarcha-AWS deleted the ml-truncation branch April 13, 2023 16:21
opensearch-trigger-bot bot pushed a commit that referenced this pull request Apr 13, 2023
* 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>
Naarcha-AWS pushed a commit that referenced this pull request Apr 13, 2023
* 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>
vagimeli pushed a commit that referenced this pull request Apr 25, 2023
* 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]>
vagimeli added a commit that referenced this pull request Apr 25, 2023
vagimeli pushed a commit that referenced this pull request May 4, 2023
* 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]>
vagimeli added a commit that referenced this pull request May 4, 2023
harshavamsi pushed a commit to harshavamsi/documentation-website that referenced this pull request Oct 31, 2023
* 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]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
4 - Doc review PR: Doc review in progress backport 2.6 PR: Backport label for 2.6
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[DOC]Need to change content for pretrained-models page.
5 participants