| --- |
| license: apache-2.0 |
| language: |
| - de |
| - en |
| base_model: |
| - patronus-studio/husky-sight-tool-type-classifier |
| pipeline_tag: text-classification |
| tags: |
| - tool-use |
| - ai-agents |
| - function-calling |
| - mcp |
| - tool-security |
| - security |
| - llm-security |
| - ai-safety |
| - ai-agent-security |
| - patronus |
| - multilingual |
| - modernbert |
| - onnx |
| - quantized |
| - edge |
| --- |
| |
| # Model Card for Husky Sight Tool Type Classifier Edge |
|
|
| **Quantized ONNX builds of [Husky Sight Tool Type Classifier](https://huggingface.co/patronus-studio/husky-sight-tool-type-classifier) for local / edge inference.** |
|
|
| This repository contains the size- and memory-optimized ONNX exports of the FP32 model. For |
| intended uses, class definitions, training data, benchmarks, and limitations, see the |
| [main model card](https://huggingface.co/patronus-studio/husky-sight-tool-type-classifier). |
|
|
| ## Variants |
|
|
| | Variant | Path | Quantization | |
| | --- | --- | --- | |
| | FP16 | `onnx/fp16/model.onnx` | FP16 weights | |
| | INT8 | `onnx/int8/model.onnx` | Post-training dynamic INT8 (MatMul/Gemm) | |
| | INT8 + INT4 embeddings | `onnx/int8_int4_embeddings/model.onnx` | Dynamic INT8 plus 4-bit weight-only token embeddings | |
|
|
| The exact quantization parameters are recorded in `onnx/quantization_manifest.json`. |
|
|
| ## Benchmark |
|
|
| Quantization parity on the unified_v3 held-out test split (n = 2914; max_length 128, CPU): |
|
|
| | Variant | Accuracy | F1 (macro) | Ξ F1 vs FP32 | |
| | --- | --- | --- | --- | |
| | FP32 reference (main repo) | 0.9568 | 0.9567 | β | |
| | FP16 | 0.9557 | 0.9554 | -0.0013 | |
| | INT8 | 0.9574 | 0.9573 | +0.0006 | |
| | INT8 + INT4 embeddings | 0.9550 | 0.9548 | -0.0019 | |
|
|
| Measured CPU latency (batch 64): ~18.89 ms/text for the INT8+INT4 build. |
|
|
| Full details in `metrics/quant_bench.json`. |
|
|
| ## Usage |
|
|
| ```python |
| from optimum.onnxruntime import ORTModelForSequenceClassification |
| from transformers import AutoTokenizer |
| |
| model_id = "patronus-studio/husky-sight-tool-type-classifier-edge" |
| tokenizer = AutoTokenizer.from_pretrained(model_id) |
| model = ORTModelForSequenceClassification.from_pretrained(model_id, subfolder="onnx/int8_int4_embeddings") |
| |
| inputs = tokenizer("...", return_tensors="pt") |
| logits = model(**inputs).logits |
| ``` |
|
|
| Post-process the logits as described in the main model card (softmax over the logits, argmax). |
|
|
| ## License |
|
|
| Released under the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0); the full |
| text is included as `LICENSE`. Derived from |
| [Husky Sight Tool Type Classifier](https://huggingface.co/patronus-studio/husky-sight-tool-type-classifier) and ultimately from |
| [jhu-clsp/mmBERT-small](https://huggingface.co/jhu-clsp/mmBERT-small) (MIT License); upstream |
| notices are retained. |
|
|
| ## Patronus Ark |
|
|
| This model is built to run inside **Patronus Ark**, Patronus' open-source on-device |
| AI-security scanning library (L1 native rules β L2 NTDB cascade β L3 transformer). |
| Ark is not publicly released yet β a repository link will be added here at launch. |
|
|
| --- |
|
|
| ## π‘οΈ Patronus Protect |
|
|
| Brought to you by [Patronus Protect](https://patronus.studio) β a local AI firewall that |
| secures every AI interaction, including prompts, tools and documents, before it reaches |
| your models. |
|
|
| Try it for free at [patronus.studio](https://patronus.studio). |
|
|