Views
No views yet
tiktoken gpt2, vocab 50304)| this run | paper | |
|---|---|---|
| Tokens / step | 65,536 | 524,288 |
| Steps | 10,000 | 10,000+ |
| Tokens seen | ~655M | ~3.18B |
| LR / min_lr | 3e-4 / 3e-5 | 3e-4 / 3e-5 |
| Warmup / decay | 200 / 10K (cosine) | 200 / 10K (cosine) |
| Weight decay | 0.1 | 0.1 |
| β1, β2 | 0.9, 0.95 | 0.9, 0.95 |
| Grad clip | 1.0 | 1.0 |
| Bias | True | True |
transformers-native model. The state_dict targets the
nanoGPT-style GPT(GPTConfig) class in this repo. To use it:1from huggingface_hub import snapshot_download
2import sys, json, torch
3from safetensors.torch import load_file
4
5local = snapshot_download(repo_id="Realmbird/mhc-781m-mhc")
6sys.path.insert(0, local)
7from model import GPT, GPTConfig
8
9with open(f"{local}/config.json") as f:
10 cfg = GPTConfig(**json.load(f))
11model = GPT(cfg)
12sd = load_file(f"{local}/model.safetensors")
13model.load_state_dict(sd)
14model.eval()