MPS (Apple Silicon) Support
dsanmart opened this issue · comments
Diego Sanmartin commented
Will there be MPS support for the Gemma models? It would enable access to a larger community.
Roger Lam commented
Took a look, a few things.
Linux / mps support looks to be in progress still pytorch/pytorch#81224 so running in a container isn't ready yet.
MPS has some limitations around complex tensors atm. Since gemma uses RoPE, it uses complex tensors and errors out if you run it locally.
Line 426 in 01062c9
Running locally with python scripts/run.py --ckpt gemma-2b-it.ckpt --variant 2b --device mps