Selected training excerpt · lines 75–76 · source SHA-256 5610fc7e8fc4
75 n_heads = enc_d_model // HEAD_DIM76 return enc_d_model, n_heads, enc_d_model // 2, n_heads