domi29's picture
tags: curated discovery keywords (remove competitor names, dedupe, add search terms)
0f425da verified
|
Raw
History Blame Contribute Delete
3.22 kB
---
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).