justinebert1 commited on
Commit
858b0e7
·
verified ·
1 Parent(s): 7b4e519

Upload README.md with huggingface_hub

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