Views
No views yet
| Params | 42M |
| Architecture | GPT-2 (12 layers, 512 embd, 8 heads) |
| Context | 512 tokens |
| Vocab | 8,000 (BPE, byte-level) |
| Pretraining | WikiText-2, 10k iters |
| Chat fine-tune | Alpaca-style instruct data (5k examples, 1.2k iters) |
| Format | ### Instruction:\n...\n### Response:\n |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tok = AutoTokenizer.from_pretrained("lattice-research/lattice-mini")
4model = AutoModelForCausalLM.from_pretrained("lattice-research/lattice-mini")
5
6prompt = "### Instruction:\nWrite a haiku about trains.\n### Response:\n"
7ids = tok(prompt, return_tensors="pt")
8out = model.generate(**ids, max_new_tokens=60, temperature=0.4)
9print(tok.decode(out[0], skip_special_tokens=True))model.safetensors — 42M params, fp32GPT2LMHeadModel (bias-free) so it loads with plain transformers.<|bos|> <|eos|> <|unk|> <|pad|>).