- Linear: normal(0, init_std) replaces kaiming_uniform_(a=sqrt(5)) - o_proj / mlp.down: init_std = 0.02 / sqrt(2 * n_layers) - MoE: expert down scaled by 1/sqrt(1/n_shared + 1/K) - Embedding: normal(0, 0.02), unchanged |
||
|---|---|---|
| .. | ||
| components | ||
| __init__.py | ||
| automodel.py | ||
| encoder.py | ||
| transformer.py | ||