README.md

metadata

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 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

pip install numpy onnxruntime tokenizers
python onnx_sample.py "How many steps today?"

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 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.

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.