File size: 1,089 Bytes
37c09b5 858b0e7 4c8863b 858b0e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | ---
license: mit
tags:
- onnx
- function-calling
- needle
- cactus
- browser
- sentencepiece
base_model: Cactus-Compute/needle
library_name: onnxruntime
---
# Needle fine-tune export
This repo contains a browser-ready ONNX export of a locally fine-tuned Needle checkpoint.
## Provenance
- Base model: [Cactus-Compute/needle](https://huggingface.co/Cactus-Compute/needle)
- Fine-tuned checkpoint: `needle_finetuned_20260608111244_50981_12_512_best.pkl`
- Fine-tuning data: UI tool-call dataset generated from the Needle demo UI
## Files
| File | Description |
|---|---|
| `encoder.onnx` | Needle encoder exported from the fine-tuned checkpoint |
| `decoder_step.onnx` | One-step decoder with KV-cache I/O |
| `needle.model` | SentencePiece tokenizer |
| `tokenizer-specials.json` | Special token IDs used by the model |
## Usage
Load the two ONNX graphs with `onnxruntime-web`, load `needle.model` with
`sentencepiece-js`, and run the encoder once followed by the decoder step in a JS loop.
## Notes
This export follows the public porting guide from `onnx-community/needle-onnx`.
|