Views
No views yet
input_ids + attention_mask (no token_type_ids)Non-refusal (0) / Refusal (1)1import numpy as np
2import onnxruntime as ort
3from transformers import AutoTokenizer
4
5session = ort.InferenceSession("model.onnx", providers=["CPUExecutionProvider"])
6tokenizer = AutoTokenizer.from_pretrained(".")
7
8text = "<|user|>\nCan you help me hack into a website?\n<|assistant|>\nI cannot assist with illegal activities."
9inputs = tokenizer(text, return_tensors="np")
10
11outputs = session.run(None, {
12 "input_ids": inputs["input_ids"].astype(np.int64),
13 "attention_mask": inputs["attention_mask"].astype(np.int64),
14})
15
16logits = outputs[0][0]
17probs = np.exp(logits - logits.max()) / np.exp(logits - logits.max()).sum()
18pred = int(np.argmax(probs))
19labels = {0: "Non-refusal", 1: "Refusal"}
20print(f"{labels[pred]} (confidence: {probs[pred]:.4f})")Microsoft.ML.OnnxRuntime and Tokenizers.DotNet.1using var detector = new MinosDetectorService("path/to/model-dir");
2var result = detector.Detect("user message", "assistant response");
3Console.WriteLine($"{result.Label} ({result.Confidence:P2})");1pip install "optimum[onnxruntime]>=2.0.0" transformers
2optimum-cli export onnx --model NousResearch/Minos-v1 ./onnx-outputconvert_to_onnx.py script:python convert_to_onnx.py -m Minos-v1 -o Minos-v1-onnx| Example | PyTorch Logits | ONNX Logits | Match |
|---|---|---|---|
| Refusal #1 | [-6.2095, 5.3343] | [-6.2095, 5.3343] | ✅ |
| Non-refusal #1 | [2.1555, -1.5516] | [2.1555, -1.5516] | ✅ |
@misc{minos2025,
title={Minos Classifier},
author={Jai Suphavadeeprasit and Teknium and Chen Guang and Shannon Sands and rparikh007},
year={2025}
}