-
Notifications
You must be signed in to change notification settings - Fork 503
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 config parameters for traced models #3456
Conversation
Signed-off-by: Naarcha-AWS <[email protected]>
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.
LGTM.
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.
Just a couple of comments. Looks good.
@@ -68,6 +68,13 @@ Field | Data type | Description | |||
| `framework_type` | string | The framework the model is using. Currently, we support `sentence_transformers` and `huggingface_transformers` frameworks. The `sentence_transformers` model outputs text embeddings directly, so ML Commons does not perform any post processing. For `huggingface_transformers`, ML Commons performs post processing by applying mean pooling to get text embeddings. See the example [`all-MiniLM-L6-v2` Huggingface model](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) for more details. | | |||
| `all_config` _(Optional)_ | string | This field is used for reference purposes. You can specify all model configurations in this field. For example, if you are using a Huggingface model, you can minify the `config.json` file to one line and save its contents in the `all_config` field. Once the model is uploaded, you can use the get model API operation to get all model configurations stored in this field. | | |||
|
|||
You can further customize a sentence transformer pre-trained model's post-processing logic with the following optional fields in the `model_config` object. |
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.
Does "... pre-trained sentence transformer model's post-processing logic ..." preserve the meaning of this?
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.
Yes. I'll adjust.
|
||
| Field | Data type | Description | | ||
| :--- | :--- | :--- | | ||
| `pooling_mode` | string | The post-process model output, either `mean`, `mean_sqrt_len`, `max`, `weightedmean`, or `cls`.| |
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.
Can't tell you why. But we typically capitalize data types: String, Boolean.
* Add config parameters for traced models Signed-off-by: Naarcha-AWS <[email protected]> * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- Signed-off-by: Naarcha-AWS <[email protected]> (cherry picked from commit fa643f9) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
* Add config parameters for traced models Signed-off-by: Naarcha-AWS <[email protected]> * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- Signed-off-by: Naarcha-AWS <[email protected]> (cherry picked from commit fa643f9) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
* Add config parameters for traced models * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- (cherry picked from commit fa643f9) 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 config parameters for traced models * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- (cherry picked from commit fa643f9) 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 config parameters for traced models Signed-off-by: Naarcha-AWS <[email protected]> * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- Signed-off-by: Naarcha-AWS <[email protected]>
* Add config parameters for traced models Signed-off-by: Naarcha-AWS <[email protected]> * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- Signed-off-by: Naarcha-AWS <[email protected]>
* Add config parameters for traced models Signed-off-by: Naarcha-AWS <[email protected]> * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md * Update model-serving-framework.md --------- Signed-off-by: Naarcha-AWS <[email protected]>
Fixes #2566
Checklist
For more information on following Developer Certificate of Origin and signing off your commits, please check here.