Instructions to use tasksource/tasksource-decider-nano with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use tasksource/tasksource-decider-nano with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="tasksource/tasksource-decider-nano")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("tasksource/tasksource-decider-nano") model = AutoModel.from_pretrained("tasksource/tasksource-decider-nano", device_map="auto") - Notebooks
- Google Colab
- Kaggle
tasksource-decider-nano
A 150M-parameter typed-decision model for the Decision Index.
It takes a state and typed questions (choice, noul) and returns a probability for every option. It does not
generate text.
#1 among models under 300M parameters on the Decision Index 0.4 (Full 19.85, October 2026).
| Decision Index 0.4, Full (private same-skills + new-domains) | 19.85 |
| Decision Index 0.3, public index | 33.51 |
| Requests | 140,171 ok, 7 unsupported (pairs above 8,192 tokens) |
| Latency, 1× NVIDIA A30 24 GB (shared), one request at a time | median 23 ms · mean 33 ms · p80 37 ms |
| Complete results | runs/tasksource-decider-nano |
Architecture
A joint cross-encoder on cross-encoder/ettin-reranker-150m-v1
(ModernBERT-style encoder, 8,192-token context). Every option is scored against the full state, its question
and the option text, like the reranker pair (state, "Question: <instructions>\nOption: <option>"), by a linear
head on the mean of the option's hidden states. The scores are softmaxed over each question's options.
The state is encoded once per request. All questions and options of a request share it through a tree attention mask: the state attends to itself, a question to the state and itself, and an option to the state, its question and itself. Rotary positions are those of the single pair, so every option gets the score it would get on its own (up to float rounding), while a request with 50 options costs one state encoding instead of 50.
Inference reads every input in full. There is no truncation: a pair above 8,192 tokens raises Unsupported.
Usage
from decider import Decider
model = Decider.from_pretrained("tasksource/tasksource-decider-nano")
model.answer(state={"review": "The battery died after a week."},
questions={"q": {"type": "choice", "instructions": "What is the sentiment?",
"criteria": {"neg": "negative", "pos": "positive"}}})
Decision Index run (kit 0.3, 9eb2dbe), from a local copy of this repository:
hf download tasksource/tasksource-decider-nano --local-dir tasksource-decider-nano
cd tasksource-decider-nano
python -m decision_index pipeline --engine decision_index_engine:DeciderEngine \
--option repo=. --edition 0.3 --out runs/tasksource-decider-nano
decider.py and decision_index_engine.py are self-contained (PyTorch and transformers only).
Training
64,000 steps, batches of 8 requests, learning rate 2e-5, inputs up to 2,048 tokens, seed 1. The loss is soft cross-entropy over each question's options. 20% of the questions are reworded with task-specific prompt styles, and 5% of the questions get a "none of the above" option. The training mix has 480,000 requests:
| Share | Source |
|---|---|
| 42% | tasksource/tasksource-jev-typed-decisions (release 9fe3dd87, balanced across tasks) |
| 30% | train splits of public datasets behind Decision Index benchmarks, e.g. MMLU, ANLI, WinoGrande, HellaSwag, ARC, CLINC150, BANKING77, Amazon ESCI, HoVer, RAGTruth, New Yorker caption contest, When2Call, ACOS, iSarcasmEval, NLI4CT, ContractNLI, GSM8K (at most 10,000 requests each) |
| 12% | tool use: Glaive, xLAM, ToolACE and Hermes function calling, ToolRet train, Home Assistant tool calls (at most 60,000 questions per source) |
| 8% | LLM routing: xRouteBench and LLMRouterBench train data |
| 8% | tasksource/synthetic-typed-decisions |
| 0.5% | LocalLLaMA/typed-decisions train (2 passes) |
No evaluation split is used, and benchmarks without a train split (for example RouterBench) are evaluation only. Every training request goes through a firewall against the Decision Index 0.3 suite: an exact match of the normalized state or of the (state, question) pair, and a template-insensitive word 6-gram content check, drop the request.
Evaluation
| Area | Skill | Raw |
|---|---|---|
| Knowledge & Reasoning | 9.7 | 31.9 |
| Language Understanding | 38.6 | 57.6 |
| Retrieval & Classification | 50.1 | 64.0 |
| Tools & Automation | 50.9 | 54.4 |
| Arts & Human Taste | 16.6 | 44.7 |
Raw index 50.38; kit 9eb2dbe, edition 0.3, 140,171 requests scored.
- Downloads last month
- 533