fix: mps memory leakage by moving gpt&emb to cpu

[fumiama]

see more at

261

https://github.com/pytorch/pytorch/issues/91368 https://stackoverflow.com/questions/74902903/pytorch-memory-leak-reference-cycle-in-for-loop