Views
No views yet
| Property | Value |
|---|---|
| Snapshot month | 2022-12 |
| Architecture | Decoder-only Transformer (GPT) |
| Layers | 52 |
| Hidden dim | 1536 |
| Attention heads | 12 |
| Vocab size | 50304 |
| Tokenizer | GPT-2 BPE |
| Position encoding | RoPE |
| Normalization | RMSNorm on Q/K + pre-norm |
| Activation | Squared ReLU |
| Weight tying | Yes (input emb ↔ lm_head) |
pip install transformers torch safetensors1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4repo_id = "Diamegs/PIT-1B-202212"
5
6tokenizer = AutoTokenizer.from_pretrained(repo_id)
7model = AutoModelForCausalLM.from_pretrained(
8 repo_id,
9 trust_remote_code=True, # required for custom architecture
10 torch_dtype=torch.bfloat16,
11)
12model = model.cuda()
13model.eval()1prompt = "In 2022, the global economy"
2
3inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
4output = model.generate(
5 **inputs,
6 max_new_tokens=200,
7 do_sample=True,
8 temperature=0.8,
9 top_p=0.95,
10 repetition_penalty=1.1,
11 pad_token_id=tokenizer.eos_token_id,
12)
13n_prompt = inputs["input_ids"].shape[1]
14print(tokenizer.decode(output[0][n_prompt:], skip_special_tokens=True))1# What does the model "know" about events before its cutoff?
2prompt = "The most important AI developments in early 2022 were"
3inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
4output = model.generate(
5 **inputs,
6 max_new_tokens=150,
7 do_sample=True,
8 temperature=0.7,
9 top_p=0.9,
10 pad_token_id=tokenizer.eos_token_id,
11)
12n_prompt = inputs["input_ids"].shape[1]
13print(tokenizer.decode(output[0][n_prompt:], skip_special_tokens=True))model.safetensors) — memory-mapped, fast to load, and safe (no arbitrary code execution).1@techreport{kelly2026pit,
2 title = {Scaling Point-in-Time Language Models},
3 author = {Kelly, Bryan T. and Malamud, Semyon and Schwab, Johannes and Xu, Teng Andrea},
4 institution = {Swiss Finance Institute},
5 type = {Research Paper},
6 number = {26-37},
7 year = {2026},
8 month = apr,
9 doi = {10.2139/ssrn.6681860},
10 url = {https://ssrn.com/abstract=6681860}
11}