lit-llama
lit-llama copied to clipboard
Be able to set custom precision for MPS accelerators
Port https://github.com/Lightning-AI/lit-parrot/pull/94 and https://github.com/Lightning-AI/lit-parrot/pull/98 to this repository