A 124M-parameter GPT-2 trained from scratch on 10B tokens of the
edu-FineWeb sample.
1huggingface-cli download Xerneas3318/gpt2-124m-edu-fineweb10b ckpt_019072.pt inference.py --local-dir .
2
3python3 inference.py \
4 --ckpt ckpt_019072.pt \
5 --prompt "Once upon a time" \
6 --num-samples 4 --max-new-tokens 100 --temperature 0.9 --top-k 50
The script primes on the real prompt tokens and grows the context autoregressively,
with temperature + top-k sampling. Runs on GPU (CUDA) by default, CPU with --device cpu.