elephant-1
A typed decision model: a full fine-tune of Qwen/Qwen3.5-9B that answers every question of a request with a probability per option from one shared-prefix prefill, without generating text.
Run the Decision Index
On one CUDA GPU with pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel:
git clone https://huggingface.co/baselabs/elephant-1 && cd elephant-1
bash setup.sh /tmp/w
python -m decision_index pipeline --engine engine:DecisionModel --option model=. --option device=cuda:0 \
--suite-dir <suite-0.3> --out runs/elephant-1
setup.sh installs the runtime and the Decision Index kit, pinned by constraints.txt. engine.py is the kit engine.
All rights reserved. Access is granted for Decision Index evaluation only.
- Downloads last month
- 8
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support