Skip to content

Commit

Permalink
Model updates and cleanup following upgrade to to triton 24.09
Browse files Browse the repository at this point in the history
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]>
  • Loading branch information
AnuradhaKaruppiah committed Nov 1, 2024
1 parent 0a17204 commit 943a2c9
Show file tree
Hide file tree
Showing 8 changed files with 6 additions and 136 deletions.
3 changes: 3 additions & 0 deletions models/phishing-models/README.md
Git LFS file not shown
Binary file modified models/ransomware-models/ransomw-model-long-rf/checkpoint.tl
Binary file not shown.
Binary file modified models/ransomware-models/ransomw-model-medium-rf/checkpoint.tl
Binary file not shown.
3 changes: 3 additions & 0 deletions models/sid-models/README.md
Git LFS file not shown
28 changes: 0 additions & 28 deletions models/triton-model-repo/phishing-bert-trt/1/README.md

This file was deleted.

36 changes: 0 additions & 36 deletions models/triton-model-repo/phishing-bert-trt/config.pbtxt

This file was deleted.

35 changes: 0 additions & 35 deletions models/triton-model-repo/sid-minibert-trt/1/README.md

This file was deleted.

37 changes: 0 additions & 37 deletions models/triton-model-repo/sid-minibert-trt/config.pbtxt

This file was deleted.

0 comments on commit 943a2c9

Please sign in to comment.