Views
No views yet
| Architecture | Transformer (causal LM) |
| Parameters | 1.99M |
| Layers | 8 |
| Heads | 4 |
| Embedding dim | 128 |
| Context window | 256 tokens |
| Vocab size | 1500 (BPE) |
| Training iters | 1800 |
| Best val loss | 4.4794 |
1from modeling_randygpt import RandyGPTConfig, RandyGPTForCausalLM
2from tokenizer_randygpt import RandyGPTTokenizer
3from safetensors.torch import load_file
4import torch
5
6# Load
7cfg = RandyGPTConfig.from_pretrained("MonumentalSystems/randygpt-s")
8model = RandyGPTForCausalLM(cfg)
9state = load_file("model.safetensors")
10model.load_state_dict(state, strict=True)
11model.eval()
12
13tok = RandyGPTTokenizer.from_file("tokenizer.json")
14
15# Generate
16prompt = "Once upon a time"
17ids = torch.tensor([tok.encode(prompt)], dtype=torch.long)
18out_ids = model.generate_text(ids, max_new_tokens=200, temperature=0.8)
19print(tok.decode(out_ids[0].tolist()))