Views
No views yet
The Chinchilla Laws (2022)| Property | Value |
|---|---|
| Architecture | GPT-2 (decoder-only transformer) |
| Parameters | ~90-100M |
| Context length | 256 tokens |
| Embedding dim | 768 |
| Layers | 12 |
| Attention heads | 12 |
| Vocabulary | 50,257 (GPT-2 BPE) |
| Training data | FineWeb (HuggingFaceFW/fineweb) + Custom dataset |
| Training docs | 256,000,000 tokens |
| Epochs | 1 |
| Precision | fp16 |
| Hyperparameter | Value |
|---|---|
| Optimiser | AdamW |
| Learning rate | 3e-4 |
| LR schedule | Cosine decay |
| Warmup steps | 2,000 |
| Weight decay | 0.1 |
| Effective batch size | 32 |
| Gradient accumulation | 8 steps |
| Dropout | 0.1 (residual, embedding, attention) |
1from transformers import GPT2LMHeadModel, AutoTokenizer, pipeline
2
3model = GPT2LMHeadModel.from_pretrained("helloadhavan/llara1.0-100M-base")
4tokenizer = AutoTokenizer.from_pretrained("helloadhavan/llara1.0-100M-base")
5
6gen = pipeline("text-generation", model=model, tokenizer=tokenizer)
7
8output = gen(
9 "The history of artificial intelligence",
10 max_new_tokens=200,
11 do_sample=True,
12 temperature=0.8,
13 top_p=0.95,
14 repetition_penalty=1.1,
15)
16
17print(output[0]["generated_text"])Trainer on a single GPU.Note: i am a AI hobbyist, not an AI engineer