Views
No views yet
| Params | ~125.8M (tied embeddings) |
| Layers | 12 |
| Hidden size | 768 |
| Heads / KV heads | 12 / 12 (MHA) |
| Context length | 1,024 |
| Vocab | 16,384 (custom BPE) |
| Activation | SwiGLU (silu) |
| Position | RoPE (theta 10000) |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4tok = AutoTokenizer.from_pretrained("analyticspro/slm-125m-base")
5model = AutoModelForCausalLM.from_pretrained("analyticspro/slm-125m-base")
6model.eval()
7
8ids = tok("The court held that", return_tensors="pt")
9out = model.generate(**ids, max_new_tokens=120, do_sample=True,
10 temperature=0.8, top_p=0.95,
11 pad_token_id=tok.eos_token_id)
12print(tok.decode(out[0], skip_special_tokens=True))