Views
No views yet
hobby-rs) to run it on a laptop CPU.SYSTEM: / USER: / ASSISTANT: turn format, and decode with a repetition penalty ≈1.3 (this is what tames the small-model repetition tendency).| Component | Value |
|---|---|
| Total parameters | ~500M (only a fraction is active per token) |
| Hidden size / layers | 768 / 16 (first FFN dense, the rest MoE) |
| Routed experts / active | 36 / top-6 (+ 1 always-on shared expert) |
| Attention | GQA, 12 query / 3 KV heads, decoupled head-dim 128, per-head QK-norm |
| Router | sigmoid gating, DeepSeek-V3 aux-loss-free load balancing, no top-k renorm |
| Positional | RoPE (θ up to 1e6 for the 8k-context checkpoints) |
| Tokenizer | GPT-2 byte-level BPE (50,304 vocab, sentinel-padded) |
| Optimizer | Muon on the 2-D + per-expert matrices, AdamW on everything else |
| Task | HobbyLM-Chat | HobbyLM-Base |
|---|---|---|
| ARC-challenge | 23.8 | 22.4 |
| ARC-easy | 42.2 | 42.8 |
| HellaSwag | 39.5 | 41.6 |
| PIQA | 67.1 | 69.5 |
| WinoGrande | 53.6 | 51.3 |
| OpenBookQA | 27.2 | 29.8 |
| BoolQ | 44.4 | 51.0 |
| Average | 42.5 | 44.0 |
How these were measured. All language-model scores are 0-shot through our own port of EleutherAI'slm-evaluation-harness(a customMoELMWrapperthat runs log-likelihood scoring over the HobbyLM MoE + GPT-2 tokenizer). Reference models in the comparison table were run through the identical harness and task set, so the numbers are apples-to-apples with ours — they are not copied from other model cards. We validated the harness against published cards (e.g. TinyLlama 52.75 vs card 52.99). These are small research models: read the numbers in context, not as leaderboard claims.
transformers AutoModel for it, so load it with
the small reference implementation from the GitHub repo:1# HobbyLM is a CUSTOM sparse-MoE architecture, so load it with the reference implementation —
2# NOT transformers.AutoModelForCausalLM (there is no AutoModel mapping for this arch).
3# pip install torch safetensors tiktoken huggingface_hub
4# git clone https://github.com/harishsg993010/HobbyLM && cd HobbyLM
5
6import json, torch, tiktoken
7from huggingface_hub import hf_hub_download
8from safetensors.torch import load_file
9from hobbylm.config import ModelConfig
10from hobbylm.model import MoETransformer
11from hobbylm.generate import generate
12
13repo = "rootxhacker/HobbyLM-Chat"
14cfg = ModelConfig(**{k: v for k, v in json.load(open(hf_hub_download(repo, "config.json"))).items() if k != "preset"})
15device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16cfg.expert_backend = "grouped" if device.type == "cuda" else "bmm"
17
18model = MoETransformer(cfg).to(device).eval()
19model.load_state_dict(load_file(hf_hub_download(repo, "model.safetensors")))
20
21enc = tiktoken.get_encoding("gpt2")
22prompt = "USER: Give me three tips for better sleep.\nASSISTANT:"
23ids = torch.tensor([enc.encode_ordinary(prompt)], device=device)
24out = generate(model, ids, max_new_tokens=64, temperature=0.7, top_k=0, device=device,
25 repetition_penalty=1.3) # temperature=0.0 for greedy
26print(enc.decode(out[0].tolist()))Prompt it with the trainedUSER:/ASSISTANT:turn format (a leadingSYSTEM:turn is optional). A repetition penalty around 1.3 is recommended.
hobbylm) live in rootxhacker/HobbyLM-gguf. They load
directly in the from-scratch hobby-rs CPU engine — stock llama.cpp won't load them without registering
the hobbylm architecture first.hobby-rs --model HobbyLM-Chat.gguf --prompt "..." --n 64