Compactbot commited on
Commit
dbe867d
·
1 Parent(s): 48f8c37

Add model card (#3)

Browse files

- Add model card (8e02dc4fbbc0905c5ffba19ab322606634087628)

Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # SuperSmallJokeClaude
2
+
3
+ A 1.64M-parameter tiny transformer trained from scratch on Fable 5 Claude Code assistant traces.
4
+
5
+ ## Specs
6
+ - **Params:** 1,644,032 (1.64M)
7
+ - **Architecture:** 4-layer transformer, d_model=128, 4 heads, d_ff=256, tied embeddings
8
+ - **Vocab:** BPE-8192 (from Claude Code tokenizer)
9
+ - **Context:** 512 tokens
10
+ - **Training data:** 24,249 tokens (Fable 5 Claude Code assistant traces)
11
+ - **Training:** 2000 steps, batch 8, seq 256, AdamW lr=3e-4 cosine schedule
12
+ - **Final loss:** 2.90
13
+ - **Training time:** 7 seconds on RTX 5090
14
+
15
+ ## What it does
16
+
17
+ It's a joke — a tiny model trained on tiny data. It will produce degenerate output (repeated tokens, gibberish). It exists to demonstrate that you can train a working transformer in 7 seconds on a single GPU.
18
+
19
+ ## Files
20
+ - `final.pt` — full checkpoint (model state dict + config + training metadata)
21
+ - `tokenizer.json` — BPE-8192 tokenizer (Claude Code)
22
+
23
+ ## Load it
24
+
25
+ ```python
26
+ import torch
27
+ from tokenizers import Tokenizer
28
+
29
+ ckpt = torch.load('final.pt', map_location='cpu')
30
+ config = ckpt['config']
31
+
32
+ # Build model (see train_jokeclaude_v5.py for the class definition)
33
+ model = Transformer(**config)
34
+ model.load_state_dict(ckpt['model_state_dict'])
35
+ model.eval()
36
+
37
+ tokenizer = Tokenizer.from_file('tokenizer.json')
38
+ ```
39
+
40
+ ## Disclaimer
41
+ This is an experimental/fun artifact. The model is not useful for any real task. It exists because someone asked for a "super small joke claude" and we said why not.