Views
No views yet
| Property | Value |
|---|---|
| Snapshot month | 2024-12 |
| Architecture | Decoder-only Transformer (GPT) |
| Layers | 20 |
| Hidden dim | 4096 |
| Attention heads | 32 |
| 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 = "anonymneurips2027/PIT-4B-FT-201312"
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 2024, 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 2024 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).