6 tokens

Color Legend

Attention
Feed-Forward
Layer Norm
Embeddings
Residual / Skip
Softmax / Output