bcui19's picture
Change `wte` to use shared embedding
5fdc9e6