samcheng0 commited on
Commit
50a4dab
·
verified ·
1 Parent(s): a947587

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +24 -11
README.md CHANGED
@@ -2,23 +2,36 @@
2
  language: en
3
  tags:
4
  - tiny
5
- - reasoning
6
- - alibi
7
- - gqa
 
8
  ---
9
 
10
- # Lumia Tiny (V2)
11
 
12
- Custom PyTorch reasoning SFT model. ~1.1M params.
13
 
14
  ## Architecture
15
- - ALiBi + GQA (8 heads, 4 KV heads)
16
- - RMSNorm + SiLU gating
17
- - Tied embeddings, head_dim=16
18
- - Vocab: 1757 (no filler)
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  ## Usage
21
  ```python
22
- from scripts.model_tiny import TinyModel
23
- model = TinyModel() # V2 defaults
24
  ```
 
2
  language: en
3
  tags:
4
  - tiny
5
+ - pct-v3
6
+ - rpw
7
+ - gpp
8
+ - vcr
9
  ---
10
 
11
+ # Lumia Tiny (PCT-V3)
12
 
13
+ Custom PyTorch LM ~969K params. Arsitektur dari first principles.
14
 
15
  ## Architecture
16
+ - **RPW** Relative Positional Warp (learned Fourier additive bias)
17
+ - **GQA** 8 query / 4 KV heads
18
+ - **GPP** Gated Principal Projection (96-dim bottleneck)
19
+ - **VCR** Variance-Controlled Residual ( gating)
20
+ - **ECI** — Entropy-Calibrated Initialization
21
+ - Vocab: 4096, RMSNorm, tied embeddings, 6 layers
22
+
23
+ ## Files
24
+ | File | Description |
25
+ |---|---|
26
+ | `model_tiny.py` | Arsitektur PCT-V3 (RPW, GQA, GPP, VCR, TinyModel) |
27
+ | `train_tiny.py` | Training loop (IterableDataset, cosine LR, checkpoint) |
28
+ | `train_tiny.yaml` | Konfigurasi training (LR 3e-4, batch 8, GA 4, 50k steps) |
29
+ | `config.json` | HF config auto-map untuk TinyModel |
30
+ | `tokenizer.json` | BPE tokenizer 4096 vocab |
31
+ | `gen_tokenizer.py` | Generator tokenizer |
32
 
33
  ## Usage
34
  ```python
35
+ from model_tiny import create_model
36
+ model = create_model() # PCT-V3 defaults
37
  ```