vocab_size × emb_dim
50,257 × 768

        __
        __