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

Model updates and cleanup following upgrade to to triton 24.09 #2036

Merged

Conversation

AnuradhaKaruppiah
Copy link
Contributor

@AnuradhaKaruppiah AnuradhaKaruppiah commented Nov 1, 2024

Closes #2025

  1. Removed unused models - sid-minibert-trt and phishing-bert-trt
  2. Update ransomware-models - ransomw-model-long-rf and ransomw-model-medium-rf with ones regenerated using treelite 4.3

With this change all pretrained models can be loaded - +-------------------------+---------+--------+
| Model | Version | Status |
+-------------------------+---------+--------+
| abp-nvsmi-xgb | 1 | READY |
| abp-pcap-xgb | 1 | READY |
| all-MiniLM-L6-v2 | 1 | READY |
| log-parsing-onnx | 1 | READY |
| phishing-bert-onnx | 1 | READY |
| ransomw-model-long-rf | 1 | READY |
| ransomw-model-medium-rf | 1 | READY |
| ransomw-model-short-rf | 1 | READY |
| root-cause-binary-onnx | 1 | READY |
| sid-minibert-onnx | 1 | READY |
+-------------------------+---------+--------+

Description

Closes

By Submitting this PR I confirm:

  • I am familiar with the Contributing Guidelines.
  • When the PR is ready for review, new or existing tests cover these changes.
  • When the PR is ready for review, the documentation is up to date with these changes.

1. Removed unused models - sid-minibert-trt and phishing-bert-trt
2. Update ransomware-models - ransomw-model-long-rf and
   ransomw-model-medium-rf with ones regenerated using treelite 4.3

With this change all pretrained models can be loaded -
+-------------------------+---------+--------+
| Model                   | Version | Status |
+-------------------------+---------+--------+
| abp-nvsmi-xgb           | 1       | READY  |
| abp-pcap-xgb            | 1       | READY  |
| all-MiniLM-L6-v2        | 1       | READY  |
| log-parsing-onnx        | 1       | READY  |
| phishing-bert-onnx      | 1       | READY  |
| ransomw-model-long-rf   | 1       | READY  |
| ransomw-model-medium-rf | 1       | READY  |
| ransomw-model-short-rf  | 1       | READY  |
| root-cause-binary-onnx  | 1       | READY  |
| sid-minibert-onnx       | 1       | READY  |
+-------------------------+---------+--------+

Signed-off-by: Anuradha Karuppiah <[email protected]>
Signed-off-by: Tad ZeMicheal <[email protected]>
@AnuradhaKaruppiah AnuradhaKaruppiah added bug Something isn't working non-breaking Non-breaking change labels Nov 1, 2024
@AnuradhaKaruppiah AnuradhaKaruppiah requested a review from a team as a code owner November 1, 2024 16:58
Copy link

copy-pr-bot bot commented Nov 1, 2024

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@AnuradhaKaruppiah
Copy link
Contributor Author

/ok to test

@dagardner-nv
Copy link
Contributor

The two README.md files shouldn't be in LFS (is it something in the path?)

@AnuradhaKaruppiah AnuradhaKaruppiah force-pushed the test-release-1 branch 2 times, most recently from afc94d7 to 8a86bbd Compare November 1, 2024 18:38
@dagardner-nv
Copy link
Contributor

/ok to test

@dagardner-nv
Copy link
Contributor

/ok to test

@dagardner-nv
Copy link
Contributor

/merge

@rapids-bot rapids-bot bot merged commit 6f5e325 into nv-morpheus:branch-24.10 Nov 1, 2024
12 of 13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working non-breaking Non-breaking change
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

3 participants