| ---
|
| license: other
|
| license_name: personal-use-free-commercial-paid
|
| license_link: LICENSE
|
| tags:
|
| - onnx
|
| - smartwatch
|
| - intent-classification
|
| - text-generation
|
| - wearable
|
| library_name: onnxruntime
|
| pipeline_tag: text-generation
|
| ---
|
|
|
| # Smartwatch LM v0.2
|
|
|
| Exported from [collab-run-2](../collab-run-2) training. Small GPT for wrist-wearable chat with intent tags like `<INTENT:GET_STEPS>`.
|
|
|
| ## Model details
|
|
|
| | Property | Value |
|
| |----------|-------|
|
| | Architecture | 6-layer causal GPT (**~15.4M params**) |
|
| | Context length | 256 tokens |
|
| | Vocab size | 5533 (BPE) |
|
| | Best val loss | 0.3243 |
|
| | Training data | tinydata.txt, deepdata.txt, tinydata1.txt, data1.txt, data3.txt |
|
| | Export version | 0.2 |
|
|
|
| ## Files
|
|
|
| | File | Purpose |
|
| |------|---------|
|
| | `smartwatch_lm_merged.onnx` | On-device inference (ONNX Runtime, opset 17, ~0 MB) |
|
| | `checkpoint.pt` | PyTorch weights |
|
| | `tokenizer.json` | BPE tokenizer |
|
| | `config.json` | Architecture + ONNX I/O |
|
| | `model.py` / `chat.py` | PyTorch load + REPL |
|
| | `reply_utils.py` | BPE cleanup, intent parse, slot fill |
|
| | `onnx_sample.py` | ONNX generate sample |
|
|
|
| ## Quick start
|
|
|
| ```bash
|
| pip install numpy onnxruntime tokenizers
|
| python onnx_sample.py "How many steps today?"
|
| ```
|
|
|
| ```bash
|
| pip install torch tokenizers
|
| python chat.py
|
| ```
|
|
|
| ## ONNX I/O
|
|
|
| - **Input:** `input_ids` int64 `[batch, seq]` (max seq = 256)
|
| - **Output:** `logits` float `[batch, seq, vocab_size]`
|
|
|
| See [Export-0.1 docs](docs/) for gibberish cleanup, intents, and device integration.
|
|
|
| ## License
|
|
|
| **Personal use is free.** You may use, modify, and share this model for non-commercial, personal, educational, and research purposes under the terms in [LICENSE](LICENSE).
|
|
|
| **Commercial use requires a paid license.** If you want to use this model in a product, service, or other commercial context, contact the repository maintainer to obtain commercial licensing terms.
|
|
|
